23b719w commited on
Commit
c47f9ef
·
1 Parent(s): 24cff10

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -120,7 +120,7 @@ def video_fn(video_in_filepath):
120
  video_writer.release()
121
  cv2.destroyAllWindows()
122
  cv2.waitKey(1)
123
- return video_writer
124
 
125
  REPO_ID = "23b719w/assignment2_tfodmodel"
126
  detection_model = load_model()
 
120
  video_writer.release()
121
  cv2.destroyAllWindows()
122
  cv2.waitKey(1)
123
+ return cv2.VideoCapture(video_out_filepath)
124
 
125
  REPO_ID = "23b719w/assignment2_tfodmodel"
126
  detection_model = load_model()