Upload app.py

app.py

@@ -100,7 +100,7 @@ def compute_correlations_faiss(index: faiss.IndexFlatIP, dataset,
 
 
     corr_df = pd.DataFrame({
-        'book': [dataset.loc(i) for i in I[0]],
+        'book': [dataset.index(i) for i in I[0]],
         'corr': similarities[0]
     })
     return corr_df.sort_values('corr', ascending=False)