fix: load model from HF

app.py

@@ -5,11 +5,22 @@ import cv2
 import streamlit as st
 
 from ultralytics import YOLO
+from huggingface_hub import hf_hub_url, cached_download
 
 
 @st.cache_resource
 def load_model():
-    return YOLO("models/soccer_ball.pt")
+    repo_id = "BreakIntoData/cv_workshop"
+    filename = "soccer_ball.pt"
+
+    # Create a URL for the model file on the Hugging Face Hub
+    model_url = hf_hub_url(repo_id, filename)
+
+    # Download the model file from the Hub and cache it locally
+    cached_model_path = cached_download(model_url)
+
+    # Load the model using YOLO from the cached model file
+    return YOLO(cached_model_path)
 
 
 def process_video(video_path):

requirements.txt

@@ -1,4 +1,5 @@
-ultralytics
-opencv-python
-numpy
-streamlit
+ultralytics==8.2.63
+opencv-python==4.10.0.84
+numpy==1.26.4
+streamlit==1.37.1
+huggingface_hub==0.24.5