Update app.py

app.py

@@ -54,18 +54,18 @@ def main():
         # Get paper IDs
         _, I = faiss_index.search(np.array(vector).astype("float32"), k=num_results)
         
-        answers=row['answers']['text']
-        answers_URLs = row['answers_urls']['url']
-        for k in range(len(answers_URLs)):
-            answers = [answer.replace(f'_URL_{k}_',answers_URLs[k]) 
-            for answer in answers]
-
-        frame = data
         # Get individual results
         for id_ in I.flatten().tolist():
-            f = frame[id_]
+            row = data[id_]
+            
+            answers=row['answers']['text']
+            answers_URLs = row['answers_urls']['url']
+            for k in range(len(answers_URLs)):
+                answers = [answer.replace(f'_URL_{k}_',answers_URLs[k]) for answer in answers]
+            
+            
             st.write(
-                f"""**Title**: {f['title']}
+                f"""**Title**: {row['title']}
                 **Top Answer**: {answers[0]}
                 """
             )