ayad33 commited on
Commit
ac37278
·
verified ·
1 Parent(s): a7a77f0

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
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 = False, exist_ok = True)
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