ayoubsa commited on
Commit
9e13fc9
·
verified ·
1 Parent(s): ff8d4cb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,8 +4,8 @@ from ultralytics import YOLO
4
  import gradio as gr
5
 
6
  # Load the YOLO model
7
- #MODEL_URL = 'https://huggingface.co/ayoubsa/yolo_model/resolve/main/best.pt'
8
- MODEL_URL = 'best.pt'
9
  model = YOLO(MODEL_URL)
10
 
11
  # Define the prediction function for Gradio
 
4
  import gradio as gr
5
 
6
  # Load the YOLO model
7
+ MODEL_URL = 'https://huggingface.co/ayoubsa/yolo_model/resolve/main/best.pt'
8
+ #MODEL_URL = 'best.pt'
9
  model = YOLO(MODEL_URL)
10
 
11
  # Define the prediction function for Gradio