Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
|
@@ -114,7 +114,7 @@ async def predict_video(file: UploadFile = File(...)):
|
|
| 114 |
|
| 115 |
# Perform inference on the video
|
| 116 |
|
| 117 |
-
model.predict('video.mp4', save = True, device = 'cpu', project = 'results', name = 'avi', verbose =
|
| 118 |
# model.predict("video.mp4", device = 'cuda', verbose = False)
|
| 119 |
|
| 120 |
# Return the video file
|
|
|
|
| 114 |
|
| 115 |
# Perform inference on the video
|
| 116 |
|
| 117 |
+
model.predict('video.mp4', save = True, device = 'cpu', project = 'results', name = 'avi', verbose = True, exist_ok = True)
|
| 118 |
# model.predict("video.mp4", device = 'cuda', verbose = False)
|
| 119 |
|
| 120 |
# Return the video file
|