mudaza commited on
Commit
cf0fcac
·
1 Parent(s): aaf24e8

modified code and add files

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from pydantic import BaseModel
6
  from fastapi.middleware.cors import CORSMiddleware
7
  import torch
8
 
9
- corpus = pickle.load(open("./corpus/all_embeddings_disease.pickle", "rb"))
10
  # label_encoder = pickle.load(open("./corpus/label_encoder.pickle", "rb"))
11
  # model = SentenceTransformer("sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2")
12
  model = SentenceTransformer('sentence-transformers/paraphrase-multilingual-mpnet-base-v2')
 
6
  from fastapi.middleware.cors import CORSMiddleware
7
  import torch
8
 
9
+ corpus = pickle.load(open("./corpus/all_embeddings_disease.pickle", "rb")).astype("float")
10
  # label_encoder = pickle.load(open("./corpus/label_encoder.pickle", "rb"))
11
  # model = SentenceTransformer("sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2")
12
  model = SentenceTransformer('sentence-transformers/paraphrase-multilingual-mpnet-base-v2')