warleagle commited on
Commit
4cdf93f
·
verified ·
1 Parent(s): e32f802

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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("data/symptom_checker/top_150_symps_by_spec.json", 'r') as f:
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)