Update app.py
Browse files
app.py
CHANGED
@@ -39,7 +39,7 @@ vectorizer = TfidfVectorizer(stop_words='english')
|
|
39 |
tfidf_matrix = vectorizer.fit_transform(movies_df['summary'])
|
40 |
|
41 |
# Define function to retrieve similar movies based on a query
|
42 |
-
def retrieve_similar_movies(query, df, tfidf_matrix, top_n=
|
43 |
query_vec = vectorizer.transform([query])
|
44 |
cosine_similarities = cosine_similarity(query_vec, tfidf_matrix).flatten()
|
45 |
top_indices = cosine_similarities.argsort()[-top_n:][::-1]
|
|
|
39 |
tfidf_matrix = vectorizer.fit_transform(movies_df['summary'])
|
40 |
|
41 |
# Define function to retrieve similar movies based on a query
|
42 |
+
def retrieve_similar_movies(query, df, tfidf_matrix, top_n=10):
|
43 |
query_vec = vectorizer.transform([query])
|
44 |
cosine_similarities = cosine_similarity(query_vec, tfidf_matrix).flatten()
|
45 |
top_indices = cosine_similarities.argsort()[-top_n:][::-1]
|