Update app.py

app.py

@@ -49,19 +49,25 @@ def query(texts):
 
 output = (dict(inputs = texts))
 
+print("output done")
+
 embeddings = pd.DataFrame(output)
 embeddings.to_csv("embeddings.csv", index=False)
 
+print("embeddings done")
+
 # 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)
 
+print("dataset_embeddings done")
 # embeddings_new = pd.read_csv(embeddings.csv)
 # dataset_embeddings = torch.from_numpy(embeddings_new.to_pandas().to_numpy()).to(torch.float)
 
 question = ["How can Medicare help me?"]
 output = query(question)
 
+print("output done")
 
 query_embeddings = torch.FloatTensor(output)
 print(f"The size of our embedded dataset is {dataset_embeddings.shape} and of our embedded query is {query_embeddings.shape}.")