Gopala Krishna
commited on
Commit
·
76ecbf9
1
Parent(s):
e298fa8
Trying with Movie Name
Browse files- .vs/MovieRecommendations/v17/.wsuo +0 -0
- 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(
|
|
|
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 |
-
|
|
|
62 |
iface = gr.Interface(
|
63 |
fn=recommend_movies,
|
64 |
-
inputs=
|
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."
|