Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -3,8 +3,8 @@ import cv2
|
|
3 |
import numpy as np
|
4 |
import torch
|
5 |
|
6 |
-
# Load the YOLOv7 model
|
7 |
-
model = torch.hub.load('WongKinYiu/yolov7', 'yolov7')
|
8 |
|
9 |
def detect_objects(image):
|
10 |
img = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
|
@@ -24,7 +24,6 @@ def detect_objects(image):
|
|
24 |
return annotated_image
|
25 |
|
26 |
def detect_live_objects(video):
|
27 |
-
# Capture frames from the video feed and process them
|
28 |
img = cv2.cvtColor(video, cv2.COLOR_RGB2BGR)
|
29 |
results = model(img)
|
30 |
|
|
|
3 |
import numpy as np
|
4 |
import torch
|
5 |
|
6 |
+
# Load the YOLOv7 model with force_reload to ensure fresh download
|
7 |
+
model = torch.hub.load('WongKinYiu/yolov7', 'yolov7', force_reload=True)
|
8 |
|
9 |
def detect_objects(image):
|
10 |
img = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
|
|
|
24 |
return annotated_image
|
25 |
|
26 |
def detect_live_objects(video):
|
|
|
27 |
img = cv2.cvtColor(video, cv2.COLOR_RGB2BGR)
|
28 |
results = model(img)
|
29 |
|