Arghya Ghosh commited on
Commit
1d09eec
·
verified ·
1 Parent(s): 46929a1

Implement IntelliFlix: Semantic Movie Recommender app with Streamlit and model loading

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -51,7 +51,7 @@ def load_model_and_data():
51
  # Load files from Hugging Face Hub
52
  with st.spinner("Loading movie data..."):
53
  csv_path = hf_hub_download(
54
- repo_id=REPO_ID,
55
  filename="data/tmdb_movies_dataset_processed.csv",
56
  repo_type="dataset",
57
  cache_dir='/tmp/huggingface'
@@ -60,7 +60,7 @@ def load_model_and_data():
60
 
61
  with st.spinner("Loading embeddings..."):
62
  embeddings_path = hf_hub_download(
63
- repo_id=REPO_ID,
64
  filename="embeddings/movie_ovierview_embeddings.npy",
65
  repo_type="dataset",
66
  cache_dir='/tmp/huggingface'
@@ -69,7 +69,7 @@ def load_model_and_data():
69
 
70
  with st.spinner("Loading FAISS index..."):
71
  faiss_path = hf_hub_download(
72
- repo_id=REPO_ID,
73
  filename="indexes/movie_overview_index.faiss",
74
  repo_type="dataset",
75
  cache_dir='/tmp/huggingface'
@@ -139,6 +139,6 @@ if query:
139
  # Footer
140
  st.markdown("---")
141
  st.markdown("""
142
- **Built with** ❤️ **by [Arghya Ghosh](https://arghya.dev)**
143
  *Technologies used: FAISS + Sentence Transformers + Streamlit*
144
  """)
 
51
  # Load files from Hugging Face Hub
52
  with st.spinner("Loading movie data..."):
53
  csv_path = hf_hub_download(
54
+ repo_id=REPO_ID,
55
  filename="data/tmdb_movies_dataset_processed.csv",
56
  repo_type="dataset",
57
  cache_dir='/tmp/huggingface'
 
60
 
61
  with st.spinner("Loading embeddings..."):
62
  embeddings_path = hf_hub_download(
63
+ repo_id=REPO_ID,
64
  filename="embeddings/movie_ovierview_embeddings.npy",
65
  repo_type="dataset",
66
  cache_dir='/tmp/huggingface'
 
69
 
70
  with st.spinner("Loading FAISS index..."):
71
  faiss_path = hf_hub_download(
72
+ repo_id=REPO_ID,
73
  filename="indexes/movie_overview_index.faiss",
74
  repo_type="dataset",
75
  cache_dir='/tmp/huggingface'
 
139
  # Footer
140
  st.markdown("---")
141
  st.markdown("""
142
+ **Built with** ❤️ **by [Arghya Ghosh](https://arghya.dev)**
143
  *Technologies used: FAISS + Sentence Transformers + Streamlit*
144
  """)