Fabrice-TIERCELIN commited on
Commit
153b44f
1 Parent(s): 2299926

Fix annotation

Browse files
Files changed (1) hide show
  1. gradio_demo.py +1 -0
gradio_demo.py CHANGED
@@ -201,6 +201,7 @@ def stage2_process(
201
 
202
  return result_slider, result_gallery, restore_information, event_id
203
 
 
204
  def restore(
205
  model,
206
  edm_steps,
 
201
 
202
  return result_slider, result_gallery, restore_information, event_id
203
 
204
+ @spaces.GPU(duration=600)
205
  def restore(
206
  model,
207
  edm_steps,