Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 from ultralytics import YOLO
+from keras.utils import get_file
 from PIL import Image
 import io
 
@@ -55,8 +56,11 @@ def load_model(model_path):
         st.error(f"Error loading model: {e}")
         return None
 
-# Path to your custom model
-MODEL_PATH = "./rssi_last.pt"
+MODEL_PATH = get_file(
+    force_download=False,
+    origin="https://huggingface.co/CineAI/RSSI/resolve/main/rssi_last.pt",
+    file_hash="c689b8a006e574bf5e4f2ff252230cb59ac3de42ca24875000144c2f552bd837"
+)
 model = load_model(MODEL_PATH)
 
 # --- Sidebar ---