coolkrishds commited on
Commit
12171ac
·
1 Parent(s): 1d6b2f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -57,7 +57,7 @@ embeddings.to_csv("embeddings.csv", index=False)
57
  # dataset_embeddings = torch.from_numpy(faqs_embeddings["train"].to_pandas().to_numpy()).to(torch.float)
58
 
59
  # dataset_embeddings = torch.from_numpy(embeddings.to_pandas().to_numpy()).to(torch.float)
60
- dataset_embeddings = torch.from_numpy(embeddings.to_numpy()).to(torch.float)
61
 
62
  question = ["How can Medicare help me?"]
63
  output = query(question)
 
57
  # dataset_embeddings = torch.from_numpy(faqs_embeddings["train"].to_pandas().to_numpy()).to(torch.float)
58
 
59
  # dataset_embeddings = torch.from_numpy(embeddings.to_pandas().to_numpy()).to(torch.float)
60
+ dataset_embeddings = torch.FloatTensor(embeddings)
61
 
62
  question = ["How can Medicare help me?"]
63
  output = query(question)