Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 =
|
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()
|