Spaces:
Sleeping
Sleeping
modified code and add files
Browse files
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')
|