Update src/streamlit_app.py

src/streamlit_app.py

@@ -14,8 +14,16 @@ from pricer import (
 # -----------------------------
 # LOAD TRAINED ML MODEL
 # -----------------------------
-ml_model = joblib.load("demand_model.pkl")
+# ml_model = joblib.load("demand_model.pkl")
+import joblib
+from huggingface_hub import hf_hub_download
+
+model_path = hf_hub_download(
+    repo_id="KB-Infinity-Tech/AIMSRICHackatonday1XGboostTrainedmodel",
+    filename="demand_model.pkl"
+)
 
+ml_model = joblib.load(model_path)
 def ml_predict(price, age_hours, comp_avg, product_encoded):
     X = pd.DataFrame([[
         price,