2384603g commited on
Commit
e8b575f
·
verified ·
1 Parent(s): 177b928

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -52,7 +52,7 @@ def process_video(video):
52
 
53
  if success:
54
  # Run YOLO inference on the frame on GPU Device 0
55
- results = model(frame, conf=0.5, device=0)
56
 
57
  # Visualize the results on the frame
58
  annotated_frame = results[0].plot()
 
52
 
53
  if success:
54
  # Run YOLO inference on the frame on GPU Device 0
55
+ results = detection_model.predict(frame, conf=0.5)
56
 
57
  # Visualize the results on the frame
58
  annotated_frame = results[0].plot()