Hugging Face's logo

Spaces:
Yudsky
/
pest-detection
Runtime error

App Files Community
Yudsky commited on
Commit
36e46ad
·
verified ·
1 Parent(s): 4280490

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ import os
9
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
10
 
11
  # Step 1: Download the YOLO model from Hugging Face
12
- url = 'https://huggingface.co/Yudsky/pest-detection-yolo11/resolve/main/best.pt'
13
  model_path = 'best.pt'
14
 
15
  # Check if the model file already exists, if not, download it
 
9
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
10
 
11
  # Step 1: Download the YOLO model from Hugging Face
12
+ url = 'https://huggingface.co/Yudsky/pest-detection-yolo11/resolve/main/best1.pt?download=true'
13
  model_path = 'best.pt'
14
 
15
  # Check if the model file already exists, if not, download it