Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,7 +8,7 @@ import os
|
|
| 8 |
os.environ['YOLO_CONFIG_DIR'] = '/tmp'
|
| 9 |
|
| 10 |
# Initialize the YOLO model
|
| 11 |
-
model = YOLO('
|
| 12 |
|
| 13 |
def process_video(video):
|
| 14 |
print("Processing video...")
|
|
|
|
| 8 |
os.environ['YOLO_CONFIG_DIR'] = '/tmp'
|
| 9 |
|
| 10 |
# Initialize the YOLO model
|
| 11 |
+
model = YOLO('yolov5n.pt') # Ensure yolov5n.pt is in the same directory as this script
|
| 12 |
|
| 13 |
def process_video(video):
|
| 14 |
print("Processing video...")
|