fffiloni commited on
Commit
6e159a1
1 Parent(s): e3994dc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 RuntimeError:
161
- return "error", None
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