warleagle commited on
Commit
3326af6
·
verified ·
1 Parent(s): 7b86ace

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -6,8 +6,8 @@ from sentence_transformers.util import cos_sim
6
  from sentence_transformers import SentenceTransformer
7
  import gradio as gr
8
  #%%
9
- etalon = pd.read_csv("data/symptomps_specialist_mkb_issledovania/ranker/raw_data/etalon_prod.csv")
10
- df = pd.read_csv("data/symptomps_specialist_mkb_issledovania/ranker/raw_data/preprocessed_train_classify_rec_spec_filtered_by_etalon.csv")
11
  df = df[df['is_match'] == 1]
12
 
13
  model = SentenceTransformer('sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2').to("cuda")
@@ -16,7 +16,7 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
16
 
17
  unique_complaints = df['Жалобы'].values.tolist()
18
 
19
- with open("experiments/symptom_recommender/embeddings_scripts/app/embeddings.npy", 'rb') as f:
20
  unique_complaints_embeddings_st = np.load(f)
21
 
22
  def get_recommend(user_input,
 
6
  from sentence_transformers import SentenceTransformer
7
  import gradio as gr
8
  #%%
9
+ etalon = pd.read_csv("etalon_prod.csv")
10
+ df = pd.read_csv("preprocessed_train_classify_rec_spec_filtered_by_etalon.csv")
11
  df = df[df['is_match'] == 1]
12
 
13
  model = SentenceTransformer('sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2').to("cuda")
 
16
 
17
  unique_complaints = df['Жалобы'].values.tolist()
18
 
19
+ with open("embeddings.npy", 'rb') as f:
20
  unique_complaints_embeddings_st = np.load(f)
21
 
22
  def get_recommend(user_input,