ddriscoll commited on
Commit
b9f4225
·
verified ·
1 Parent(s): a7ed780

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -27,8 +27,10 @@ else:
27
  # Load models from Hugging Face
28
  # ---------------------------
29
 
30
- # YOLO detection model – replace the URL with your actual Hugging Face model URL.
31
- yolo_model = YOLO("https://huggingface.co/your-hf-username/your-yolov8-model/resolve/main/model.pt")
 
 
32
 
33
  # QA pipeline (for Ask Eurybia)
34
  qa_pipeline = pipeline("question-answering", model="deepset/roberta-base-squad2")
 
27
  # Load models from Hugging Face
28
  # ---------------------------
29
 
30
+ # YOLO detection model – load the .pt file from a Hugging Face repo.
31
+ # Replace "your-hf-username/your-yolov8-model" and "model.pt" with your actual repo id and filename.
32
+ yolo_weights_path = hf_hub_download(repo_id="your-hf-username/your-yolov8-model", filename="model.pt")
33
+ yolo_model = YOLO(yolo_weights_path)
34
 
35
  # QA pipeline (for Ask Eurybia)
36
  qa_pipeline = pipeline("question-answering", model="deepset/roberta-base-squad2")