mudassir032 commited on
Commit
1910e89
·
verified ·
1 Parent(s): fe94c1b

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +24 -0
app.py ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ import numpy as np
3
+ import pandas as pd
4
+ from sklearn.preprocessing import StandardScaler
5
+ from sklearn.metrics.pairwise import cosine_similarity
6
+
7
+ movies_df = pd.read_csv("movies.csv")
8
+
9
+ genre_df = movies_df['genres'].str.get_dummies("|")
10
+
11
+ st.write(genre_df)
12
+
13
+ similarity_matrix = cosine_similarity(genre_df)
14
+
15
+ random_movies = movies_df['title'].sample(5)
16
+
17
+ movie = st.text_input("Search for a movie")
18
+
19
+ if not movie:
20
+ movie0 = st.button(random_movies[0])
21
+ movie1 = st.button(random_movies[1])
22
+ movie2 = st.button(random_movies[2])
23
+ movie3 = st.button(random_movies[3])
24
+ movie4 = st.button(random_movies[4])