Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ from nltk.corpus import stopwords
|
|
16 |
model = SentenceTransformer('sentence-transformers/multi-qa-distilbert-cos-v1')
|
17 |
russian_stopwords = stopwords.words('russian') + ['ВАШ']
|
18 |
|
19 |
-
with open("
|
20 |
symps = json.load(f)
|
21 |
|
22 |
with open("embeddings.npy", 'rb') as f:
|
@@ -40,7 +40,6 @@ def get_symptomps_v2(text, treshold = 0.7):
|
|
40 |
text = [text]
|
41 |
|
42 |
X = vectorizer.fit_transform(text)
|
43 |
-
# print(vectorizer.get_feature_names_out())
|
44 |
text_emb = model.encode(vectorizer.get_feature_names_out(), batch_size=64)
|
45 |
cos_sim_m = util.cos_sim(text_emb, embs).numpy()
|
46 |
cos_sim_m = np.where(cos_sim_m > treshold, cos_sim_m, -1)
|
|
|
16 |
model = SentenceTransformer('sentence-transformers/multi-qa-distilbert-cos-v1')
|
17 |
russian_stopwords = stopwords.words('russian') + ['ВАШ']
|
18 |
|
19 |
+
with open("top_150_symps_by_spec.json", 'r') as f:
|
20 |
symps = json.load(f)
|
21 |
|
22 |
with open("embeddings.npy", 'rb') as f:
|
|
|
40 |
text = [text]
|
41 |
|
42 |
X = vectorizer.fit_transform(text)
|
|
|
43 |
text_emb = model.encode(vectorizer.get_feature_names_out(), batch_size=64)
|
44 |
cos_sim_m = util.cos_sim(text_emb, embs).numpy()
|
45 |
cos_sim_m = np.where(cos_sim_m > treshold, cos_sim_m, -1)
|