Gopala Krishna commited on
Commit
76ecbf9
·
1 Parent(s): e298fa8

Trying with Movie Name

Browse files
Files changed (2) hide show
  1. .vs/MovieRecommendations/v17/.wsuo +0 -0
  2. app.py +5 -3
.vs/MovieRecommendations/v17/.wsuo CHANGED
Binary files a/.vs/MovieRecommendations/v17/.wsuo and b/.vs/MovieRecommendations/v17/.wsuo differ
 
app.py CHANGED
@@ -33,7 +33,8 @@ def find_similar_movies(movie_id, X, k, metric='cosine', show_distance=False):
33
  neighbour_ids.pop(0)
34
  return neighbour_ids
35
 
36
- def recommend_movies(movie_id):
 
37
  similar_ids = find_similar_movies(movie_id, X, k=10)
38
  movie_title = movie_titles[movie_id]
39
  recommendations = []
@@ -58,10 +59,11 @@ X, user_mapper, movie_mapper, user_inv_mapper, movie_inv_mapper = create_matrix(
58
  movie_titles = dict(zip(movies['movieId'], movies['title']))
59
 
60
  # Set up Gradio interface
61
- movie_id = gr.inputs.Number(default=3, label="Movie ID")
 
62
  iface = gr.Interface(
63
  fn=recommend_movies,
64
- inputs=movie_id,
65
  outputs="text",
66
  title="Movie Recommender System",
67
  description="Enter a movie ID and get recommendations for similar movies."
 
33
  neighbour_ids.pop(0)
34
  return neighbour_ids
35
 
36
+ def recommend_movies(movie_name):
37
+ movie_id = movies[movies['title'] == movie_name]['movieId'].values[0]
38
  similar_ids = find_similar_movies(movie_id, X, k=10)
39
  movie_title = movie_titles[movie_id]
40
  recommendations = []
 
59
  movie_titles = dict(zip(movies['movieId'], movies['title']))
60
 
61
  # Set up Gradio interface
62
+ movie_name = gr.inputs.Text
63
+
64
  iface = gr.Interface(
65
  fn=recommend_movies,
66
+ inputs=movie_name,
67
  outputs="text",
68
  title="Movie Recommender System",
69
  description="Enter a movie ID and get recommendations for similar movies."