tahirsher commited on
Commit
e433d99
·
verified ·
1 Parent(s): 366b273

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=5):
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]