Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -6,7 +6,7 @@ from sklearn.feature_extraction.text import TfidfVectorizer
|
|
| 6 |
|
| 7 |
|
| 8 |
movies_df = pd.read_csv("movies.csv")
|
| 9 |
-
|
| 10 |
genre_df = movies_df['genres'].str.get_dummies("|")
|
| 11 |
st.title("Movie Recommendation System")
|
| 12 |
@st.cache_data
|
|
@@ -53,6 +53,8 @@ if not movie:
|
|
| 53 |
movie_id = movies_df[movies_df['title'] == random_movies.iloc[id]].index
|
| 54 |
similar_movies_idx = similarity_matrix[movie_id[0]].argsort()[::-1][1:7]
|
| 55 |
similar_movies = movies_df['title'].iloc[similar_movies_idx]
|
|
|
|
|
|
|
| 56 |
with col2:
|
| 57 |
st.subheader("Recommended Movies")
|
| 58 |
for i,j in enumerate(similar_movies[:5]):
|
|
|
|
| 6 |
|
| 7 |
|
| 8 |
movies_df = pd.read_csv("movies.csv")
|
| 9 |
+
links = pd.read_csv("imdbLinks.csv")
|
| 10 |
genre_df = movies_df['genres'].str.get_dummies("|")
|
| 11 |
st.title("Movie Recommendation System")
|
| 12 |
@st.cache_data
|
|
|
|
| 53 |
movie_id = movies_df[movies_df['title'] == random_movies.iloc[id]].index
|
| 54 |
similar_movies_idx = similarity_matrix[movie_id[0]].argsort()[::-1][1:7]
|
| 55 |
similar_movies = movies_df['title'].iloc[similar_movies_idx]
|
| 56 |
+
movies_links = links.iloc[similar_movies_idx].to_numpy()
|
| 57 |
+
st.write(movie_links)
|
| 58 |
with col2:
|
| 59 |
st.subheader("Recommended Movies")
|
| 60 |
for i,j in enumerate(similar_movies[:5]):
|