AlshimaaGamalAlsaied commited on
Commit
483bdd9
1 Parent(s): 7949879
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -74,7 +74,7 @@ def video_fn(model_path, video_file, conf_thres, iou_thres, start_sec, duration)
74
  print(e)
75
  continue
76
  print("FRAME DTYPE", type(frame))
77
- out.write(model(frame, conf_thres, iou_thres))
78
  num_frames += 1
79
  print("Processed {} frames".format(num_frames))
80
  if num_frames == max_frames:
 
74
  print(e)
75
  continue
76
  print("FRAME DTYPE", type(frame))
77
+ out.write(model([frame], conf_thres, iou_thres))
78
  num_frames += 1
79
  print("Processed {} frames".format(num_frames))
80
  if num_frames == max_frames: