Hugging Face's logo

Spaces:
sharktide
/
yolo-server
Sleeping

App Files Community
sharktide commited on
Commit
151f2d9
·
verified ·
1 Parent(s): 733206d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from ultralytics import YOLO
8
  app = FastAPI()
9
 
10
  # Download YOLOv8 model from Hugging Face
11
- YOLO_MODEL_PATH = hf_hub_download(repo_id="sharktide/RDiCC", filename="best.pt")
12
 
13
  # Load YOLOv8 model
14
  yolo_model = YOLO(YOLO_MODEL_PATH)
 
8
  app = FastAPI()
9
 
10
  # Download YOLOv8 model from Hugging Face
11
+ YOLO_MODEL_PATH = hf_hub_download(repo_id="sharktide/RDiCC", filename="runs/detect/train/weights/best.pt")
12
 
13
  # Load YOLOv8 model
14
  yolo_model = YOLO(YOLO_MODEL_PATH)