Upload app.py

app.py

@@ -47,8 +47,8 @@ def get_recommendations_and_generate(query_text, k=3):
     for idx_numpy in indices[0]:
         idx = int(idx_numpy)
         trip_plan = {
-            "dest": dataset[idx]['dest'],
-            "reference_information": dataset[idx]['reference_information']
+            "dest": dataset['test']['dest'][idx],
+            "reference_information": dataset['test']['reference_information'][idx]
         }
         results.append(trip_plan)