Update src/streamlit_app.py

src/streamlit_app.py

@@ -4,17 +4,22 @@ import numpy as np
 import tensorflow as tf
 import joblib
 import os
+import zipfile
 
 # Define paths
 BASE_DIR = os.path.dirname(__file__)
-MODEL_PATH = os.path.join(BASE_DIR, "recommender_model.keras")
+ZIP_MODEL_PATH = os.path.join(BASE_DIR, "recommender_model.zip")
+EXTRACTED_MODEL_DIR = os.path.join(BASE_DIR, "recommender_model")
 MOVIES_PATH = os.path.join(BASE_DIR, "movies.csv")
 ENCODINGS_PATH = os.path.join(BASE_DIR, "encodings.pkl")
 
 @st.cache_resource
 def load_model():
     try:
-        return tf.keras.models.load_model(MODEL_PATH)
+        if not os.path.exists(EXTRACTED_MODEL_DIR):
+            with zipfile.ZipFile(ZIP_MODEL_PATH, "r") as zip_ref:
+                zip_ref.extractall(EXTRACTED_MODEL_DIR)
+        return tf.keras.models.load_model(EXTRACTED_MODEL_DIR)
     except Exception as e:
         st.error(f"Failed to load model: {e}")
         st.stop()