Spaces:
Build error
Build error
File size: 834 Bytes
ed9c848 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
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) |