mikonvergence commited on
Commit
40c3bb1
·
1 Parent(s): a1c8d5a
Files changed (1) hide show
  1. src/backend.py +5 -5
src/backend.py CHANGED
@@ -240,16 +240,16 @@ def generate_output(s2l1c_input, s2l2a_input, s1rtc_input, dem_input, num_infere
240
 
241
  if s2l2a_active and s2l1c_active and s1rtc_active and dem_active:
242
  gr.Warning("You need to remove some of the inputs that you would like to generate. If all modalities are known, there is nothing to generate.")
243
- return s2l2a_input, s2l1c_input, s1rtc_input, dem_input
244
 
245
  images=[]
246
  condition_modalities=[]
247
- if s2l2a_active:
248
- images.append(s2l2a_input)
249
- condition_modalities.append('s2_l2a')
250
  if s2l1c_active:
251
  images.append(s2l1c_input)
252
  condition_modalities.append('s2_l1c')
 
 
 
253
  if s1rtc_active:
254
  images.append(s1rtc_input)
255
  condition_modalities.append('s1_rtc')
@@ -259,7 +259,7 @@ def generate_output(s2l1c_input, s2l2a_input, s1rtc_input, dem_input, num_infere
259
 
260
  imgs_out = custom_inference(
261
  images=images,
262
- generate_modalities=[el for el in ['s2_l2a', 's2_l1c', 's1_rtc', 'dem'] if el not in condition_modalities],
263
  condition_modalities=condition_modalities,
264
  num_inference_steps=num_inference_steps_slider,
265
  seed=seed
 
240
 
241
  if s2l2a_active and s2l1c_active and s1rtc_active and dem_active:
242
  gr.Warning("You need to remove some of the inputs that you would like to generate. If all modalities are known, there is nothing to generate.")
243
+ return s2l1c_input, s2l2a_input, s1rtc_input, dem_input
244
 
245
  images=[]
246
  condition_modalities=[]
 
 
 
247
  if s2l1c_active:
248
  images.append(s2l1c_input)
249
  condition_modalities.append('s2_l1c')
250
+ if s2l2a_active:
251
+ images.append(s2l2a_input)
252
+ condition_modalities.append('s2_l2a')
253
  if s1rtc_active:
254
  images.append(s1rtc_input)
255
  condition_modalities.append('s1_rtc')
 
259
 
260
  imgs_out = custom_inference(
261
  images=images,
262
+ generate_modalities=[el for el in ['s2_l1c', 's2_l2a', 's1_rtc', 'dem'] if el not in condition_modalities],
263
  condition_modalities=condition_modalities,
264
  num_inference_steps=num_inference_steps_slider,
265
  seed=seed