mikonvergence commited on
Commit
ee37e55
·
1 Parent(s): 65328d9

zero gpu update

Browse files
Files changed (2) hide show
  1. app.py +2 -4
  2. src/utils.py +5 -1
app.py CHANGED
@@ -60,11 +60,9 @@ with gr.Blocks(theme=theme) as demo:
60
  )
61
 
62
  generate_button.click(
63
- #fn=generate_output,
64
  inputs=[s2l1c_input, s2l2a_input, s1rtc_input, dem_input, num_inference_steps_slider, guidance_scale_slider, seed_number, seed_checkbox],
65
  outputs=[s2l1c_output, s2l2a_output, s1rtc_output, dem_output],
66
  )
67
 
68
- demo.launch()
69
-
70
- demo.launch(share=True)
 
60
  )
61
 
62
  generate_button.click(
63
+ fn=generate_output,
64
  inputs=[s2l1c_input, s2l2a_input, s1rtc_input, dem_input, num_inference_steps_slider, guidance_scale_slider, seed_number, seed_checkbox],
65
  outputs=[s2l1c_output, s2l2a_output, s1rtc_output, dem_output],
66
  )
67
 
68
+ demo.queue().launch()
 
 
src/utils.py CHANGED
@@ -167,4 +167,8 @@ def sample_shuffle(interface=True):
167
  out = []
168
  for el in images:
169
  out += [el, True]
170
- return out
 
 
 
 
 
167
  out = []
168
  for el in images:
169
  out += [el, True]
170
+ return out
171
+
172
+ @spaces.GPU
173
+ def generate_output():
174
+ pass