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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -10,9 +10,8 @@ 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")
14
 
15
- device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
16
 
17
  unique_complaints = df['Жалобы'].values.tolist()
18
 
 
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')
14
 
 
15
 
16
  unique_complaints = df['Жалобы'].values.tolist()
17