mudassir032 commited on
Commit
e749a79
·
verified ·
1 Parent(s): c623b78

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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]):