Hugging Face's logo

Spaces:
ayoubsa
/
Sign_roads_detection_app
Running

App Files Community
ayoubsa commited on
Commit
8cb628c
·
verified ·
1 Parent(s): f2241a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,7 +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 = YOLO(MODEL_URL, cache_dir="/root/.cache")
9
 
10
  # 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, cache_dir="/root/.cache")
10
 
11
  # Define the prediction function for Gradio