PranavReddy18's picture
Upload 7 files
ed9c848 verified
raw
history blame contribute delete
834 Bytes
import streamlit as st
import pickle
import pandas as pd
def recomend(movie):
movie_index=movies[movies['title']==movie].index[0]
distences=similarity[movie_index]
movies_list = sorted(list(enumerate(distences)),reverse=True,key=lambda x:x[1])[1:6]
recommended_movies=[]
for i in movies_list:
recommended_movies.append(movies.iloc[i[0]].title)
return recommended_movies
movie_dict=pickle.load(open('movie_dict.pkl','rb'))
movies=pd.DataFrame(movie_dict)
similarity=pickle.load(open('similarity1.pkl','rb'))
st.title('Movie Recommender System')
selected_movie_name=st.selectbox(
'How Would you like to br contacted?',movies['title'].values
)
if st.button('Recommend'):
recommendations=recomend(selected_movie_name)
for i in recommendations:
st.write(i)