Update src/streamlit_app.py

src/streamlit_app.py

@@ -94,15 +94,20 @@ def recommend(movie):
 def load_model_files():
     """Load model files from Hugging Face Hub with caching."""
     try:
+        # Create a cache directory in the current working directory
+        cache_dir = "./.hf_cache"
+        
         # Download files from your Hugging Face repository
         movie_list_path = hf_hub_download(
-            repo_id="N4F1U/Movie_Recommender",  # Your repository ID
-            filename="movie_list.pkl"
+            repo_id="N4F1U/Movie_Recommender",
+            filename="movie_list.pkl",
+            cache_dir=cache_dir  # Specify custom cache directory
         )
         
         similarity_path = hf_hub_download(
             repo_id="N4F1U/Movie_Recommender",
-            filename="similarity.pkl"
+            filename="similarity.pkl",
+            cache_dir=cache_dir  # Specify custom cache directory
         )
         
         # Load the pickle files