Fix annotation
Browse files- 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,
|