Upload 5 files

.gitattributes

@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+datasetf.csv filter=lfs diff=lfs merge=lfs -text
+faiss_index.index filter=lfs diff=lfs merge=lfs -text

app.py

@@ -0,0 +1,71 @@
+import streamlit as st
+from PIL import Image
+import requests
+from io import BytesIO
+from sentence_transformers import SentenceTransformer
+import faiss
+import pandas as pd
+
+# import sys
+# import subprocess
+# import streamlit as st
+
+# # Debug: Print Python executable and installed packages
+# st.write(f"Python executable: {sys.executable}")
+# installed_packages = subprocess.run([sys.executable, "-m", "pip", "list"], capture_output=True, text=True).stdout
+# st.text(installed_packages)
+
+model = SentenceTransformer('cointegrated/rubert-tiny2')
+index = faiss.read_index('faiss_index.index')
+data = pd.read_csv('datasetf.csv')
+
+def vectorize(descriptions):
+    embeddings = model.encode(descriptions)
+    return embeddings
+
+
+def find_similar_shows(user_description, index, k=5):
+    query_vector = vectorize([user_description])
+    _, indices = index.search(query_vector, k)
+    return data.iloc[indices.flatten()] 
+
+def load_image(url):
+    try:
+        response = requests.get(url)
+        img = Image.open(BytesIO(response.content))
+    except Exception:
+        # If an error occurs, load the dummy image
+        img = Image.open("cat.jpg")  # Update the path to your dummy image
+    return img
+
+st.title('TV Show Recommender')
+
+# User input for the show description
+user_description = st.text_area("Describe the TV show you're looking for:")
+
+# Slider for the number of recommendations
+num_recommendations = st.slider('Number of recommendations:', min_value=1, max_value=10, value=5)
+
+# Button to get recommendations
+if st.button('Recommend') and user_description:
+    try:
+        recommended_shows = find_similar_shows(user_description, index, num_recommendations)
+
+        for idx in recommended_shows.index:
+            with st.container():
+                link = data.loc[idx, 'url']
+                poster_url = data.loc[idx, 'poster']
+                title = data.loc[idx, 'title']
+                description = data.loc[idx, 'description']
+
+                img = load_image(poster_url)  # Use the load_image function
+                
+                col1, col2 = st.columns([1, 2])
+                with col1:
+                    st.image(img, caption=title, use_column_width='always')
+                with col2:
+                    st.write(description)
+                    st.markdown(f"[More Info]({link})", unsafe_allow_html=True)
+
+    except Exception as e:
+        st.error(f"An error occurred: {e}")

datasetf.csv

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9baa1443870901e093cb11d655254adee1317f0684a5021bbc3db7f05ccb1a1b
+size 19263072

faiss_index.index

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7e9c90a44a23ff326c88c295bbd23f1615b405a52c57b659306866b71564bdaa
+size 18632685

requirements.txt

@@ -0,0 +1,105 @@
+altair==5.2.0
+asttokens==2.4.1
+attrs==23.2.0
+beautifulsoup4==4.12.3
+blinker==1.7.0
+cachetools==5.3.2
+certifi==2024.2.2
+charset-normalizer==3.3.2
+click==8.1.7
+comm==0.2.1
+debugpy==1.8.0
+decorator==5.1.1
+exceptiongroup==1.2.0
+executing==2.0.1
+faiss-cpu==1.7.4
+filelock==3.13.1
+fsspec==2024.2.0
+gitdb==4.0.11
+GitPython==3.1.41
+huggingface-hub==0.20.3
+idna==3.6
+importlib-metadata==7.0.1
+ipykernel==6.29.1
+ipython==8.21.0
+jedi==0.19.1
+Jinja2==3.1.3
+joblib==1.3.2
+jsonschema==4.21.1
+jsonschema-specifications==2023.12.1
+jupyter_client==8.6.0
+jupyter_core==5.7.1
+markdown-it-py==3.0.0
+MarkupSafe==2.1.5
+matplotlib-inline==0.1.6
+mdurl==0.1.2
+mpmath==1.3.0
+nest-asyncio==1.6.0
+networkx==3.2.1
+nltk==3.8.1
+numpy==1.26.4
+nvidia-cublas-cu12==12.1.3.1
+nvidia-cuda-cupti-cu12==12.1.105
+nvidia-cuda-nvrtc-cu12==12.1.105
+nvidia-cuda-runtime-cu12==12.1.105
+nvidia-cudnn-cu12==8.9.2.26
+nvidia-cufft-cu12==11.0.2.54
+nvidia-curand-cu12==10.3.2.106
+nvidia-cusolver-cu12==11.4.5.107
+nvidia-cusparse-cu12==12.1.0.106
+nvidia-nccl-cu12==2.19.3
+nvidia-nvjitlink-cu12==12.3.101
+nvidia-nvtx-cu12==12.1.105
+packaging==23.2
+pandas==2.2.0
+parso==0.8.3
+pexpect==4.9.0
+pillow==10.2.0
+platformdirs==4.2.0
+prompt-toolkit==3.0.43
+protobuf==4.25.2
+psutil==5.9.8
+ptyprocess==0.7.0
+pure-eval==0.2.2
+pyarrow==15.0.0
+pydeck==0.8.1b0
+Pygments==2.17.2
+python-dateutil==2.8.2
+pytz==2024.1
+PyYAML==6.0.1
+pyzmq==25.1.2
+referencing==0.33.0
+regex==2023.12.25
+requests==2.31.0
+rich==13.7.0
+rpds-py==0.17.1
+safetensors==0.4.2
+scikit-learn==1.4.0
+scipy==1.12.0
+sentence-transformers==2.3.1
+sentencepiece==0.1.99
+six==1.16.0
+smmap==5.0.1
+soupsieve==2.5
+stack-data==0.6.3
+streamlit==1.31.0
+sympy==1.12
+tenacity==8.2.3
+threadpoolctl==3.2.0
+tokenizers==0.15.1
+toml==0.10.2
+toolz==0.12.1
+torch==2.2.0
+tornado==6.4
+tqdm==4.66.1
+traitlets==5.14.1
+transformers==4.37.2
+triton==2.2.0
+typing_extensions==4.9.0
+tzdata==2023.4
+tzlocal==5.2
+urllib3==2.2.0
+validators==0.22.0
+watchdog==4.0.0
+wcwidth==0.2.13
+zipp==3.17.0