fffiloni commited on
Commit
84e6bd9
·
1 Parent(s): 6e159a1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -158,7 +158,9 @@ def run_inference(prompt, video_path, condition, video_length, seed, steps):
158
  print(f"FINISHED !")
159
  return "done", 'resized_final.mp4'
160
  except cuda.Error as e:
161
- return e, None
 
 
162
 
163
 
164
 
 
158
  print(f"FINISHED !")
159
  return "done", 'resized_final.mp4'
160
  except cuda.Error as e:
161
+ return f"CUDA Error: {e}", None
162
+ except RuntimeError as e:
163
+ return f"Runtime Error: {e}", None
164
 
165
 
166