Deepak Sahu commited on
Commit
ebf589b
·
1 Parent(s): 9062ce3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -49,8 +49,10 @@ def get_recommendation(book_title: str) -> dict:
49
  df_ranked = books_df.iloc[ranks]
50
  df_ranked = df_ranked.reset_index()
51
 
52
- books = df_ranked["book_name"][:N_RECOMMENDS]
53
  scores = similarity[ranks][:N_RECOMMENDS]
 
 
54
 
55
  return {book: score for book, score in zip(books, scores)} # referene: https://huggingface.co/docs/hub/en/spaces-sdks-gradio
56
 
 
49
  df_ranked = books_df.iloc[ranks]
50
  df_ranked = df_ranked.reset_index()
51
 
52
+ books = df_ranked["book_name"].to_list()[:N_RECOMMENDS]
53
  scores = similarity[ranks][:N_RECOMMENDS]
54
+ print(scores)
55
+ print(type(scores))
56
 
57
  return {book: score for book, score in zip(books, scores)} # referene: https://huggingface.co/docs/hub/en/spaces-sdks-gradio
58