mudassir032 commited on
Commit
5aabe52
·
verified ·
1 Parent(s): 7101501

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -66,8 +66,10 @@ else:
66
  result = search_movie(movie)
67
  id = movies_df[movies_df['title'] == result].index[0]
68
  st.write(result,id)
69
- similar_movies_idx = similarity_matrix[id].argsort()[::-1][:10]
70
- similar_movies = movies_df['title'].iloc[similar_movies_idx]
 
 
71
  st.subheader("Recommended Movies")
72
- for i,j in enumerate(similar_movies):
73
  st.write(f"\t{i+1}.{j}")
 
66
  result = search_movie(movie)
67
  id = movies_df[movies_df['title'] == result].index[0]
68
  st.write(result,id)
69
+ similar_movies_idx = similarity_matrix[id].argsort()[::-1][:5]
70
+ similar_movies = movies_df['title'].iloc[similar_movies_idx].numpy()
71
+ if result in similar_movies:
72
+ similar_moviess = np.delete(similar_movies,id)
73
  st.subheader("Recommended Movies")
74
+ for i,j in enumerate(similar_moviess):
75
  st.write(f"\t{i+1}.{j}")