Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -157,8 +157,8 @@ def run_inference(prompt, video_path, condition, video_length, seed, steps):
|
|
| 157 |
|
| 158 |
print(f"FINISHED !")
|
| 159 |
return "done", 'resized_final.mp4'
|
| 160 |
-
except
|
| 161 |
-
return
|
| 162 |
|
| 163 |
|
| 164 |
|
|
|
|
| 157 |
|
| 158 |
print(f"FINISHED !")
|
| 159 |
return "done", 'resized_final.mp4'
|
| 160 |
+
except cuda.Error as e:
|
| 161 |
+
return e, None
|
| 162 |
|
| 163 |
|
| 164 |
|