Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -136,7 +136,7 @@ from PIL import Image
|
|
136 |
import numpy as np
|
137 |
|
138 |
# Load the YOLO model
|
139 |
-
model = torch.hub.load('ultralytics/yolov5', 'custom', path='https://huggingface.co/ayoubsa/yolo_model/
|
140 |
|
141 |
# Define the prediction function
|
142 |
def predict(image):
|
|
|
136 |
import numpy as np
|
137 |
|
138 |
# Load the YOLO model
|
139 |
+
model = torch.hub.load('ultralytics/yolov5', 'custom', path='https://huggingface.co/ayoubsa/yolo_model/resolve/main/best.pt') # Replace with your uploaded model's path
|
140 |
|
141 |
# Define the prediction function
|
142 |
def predict(image):
|