ayoubsa commited on
Commit
36fca89
·
verified ·
1 Parent(s): 77b134f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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/blob/main/best.pt') # Replace with your uploaded model's path
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):