Ajay07pandey commited on
Commit
f289997
1 Parent(s): c03d967
Files changed (1) hide show
  1. app.py +36 -0
app.py ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ import pickle
3
+ import pandas as pd
4
+
5
+
6
+ st.image("Netflix.png")
7
+
8
+ movies_list = pickle.load(open("content_dict.pkl",'br'))
9
+ movies = pd.DataFrame(movies_list)
10
+
11
+ def recommend(movie):
12
+ index = movies[movies['title'] == movie].index[0]
13
+ distances = sorted(list(enumerate(similarity[index])), reverse=True, key=lambda x: x[1])
14
+ recommended_movie_names = []
15
+ for i in distances[1:6]:
16
+ recommended_movie_names.append(movies.iloc[i[0]].title)
17
+
18
+ return recommended_movie_names
19
+
20
+ # Displaying title
21
+ st.title(" Movie Recommender System ")
22
+
23
+ similarity = pickle.load(open('cosine_similarity.pkl','rb'))
24
+
25
+ movie_list = movies['title'].values
26
+ selected_movie = st.selectbox(
27
+ "Type or select a movie from the dropdown",
28
+ movie_list
29
+ )
30
+
31
+ # Setting a button
32
+ if st.button('Show Recommendation'):
33
+ recommended_movie_names = recommend(selected_movie)
34
+ st.balloons()
35
+ for j in recommended_movie_names:
36
+ st.write(j)