Update app.py

app.py

@@ -51,9 +51,11 @@ output = (dict(inputs = texts))
 embeddings = pd.DataFrame(output)
 embeddings.to_csv("embeddings.csv", index=False)
 
+# If were to upload embeddings in huggingface dataset
+# faqs_embeddings = load_dataset('ITESM/embedded_faqs_medicare')
+# dataset_embeddings = torch.from_numpy(faqs_embeddings["train"].to_pandas().to_numpy()).to(torch.float)
 
-faqs_embeddings = load_dataset('ITESM/embedded_faqs_medicare')
-dataset_embeddings = torch.from_numpy(faqs_embeddings["train"].to_pandas().to_numpy()).to(torch.float)
+dataset_embeddings = torch.from_numpy(embeddings.to_pandas().to_numpy()).to(torch.float)
 
 question = ["How can Medicare help me?"]
 output = query(question)