Yerzhxn commited on
Commit
475e44d
·
verified ·
1 Parent(s): e1433b0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,8 +5,8 @@ from transformers import BertTokenizer, BertForSequenceClassification
5
  import joblib
6
 
7
  # Загрузка модели, токенизатора и label_encoder
8
- model = torch.load("bert_model.pkl", map_location=torch.device('cpu'))
9
- tokenizer = joblib.load("bert_tokenizer.pkl")
10
 
11
  def predict_class_with_probabilities(text, model, tokenizer, max_len=128):
12
  model.eval()
 
5
  import joblib
6
 
7
  # Загрузка модели, токенизатора и label_encoder
8
+ model = torch.load("logistic_regression_model.pkl", map_location=torch.device('cpu'))
9
+ tokenizer = joblib.load("tfidf_vectorizer.pkl")
10
 
11
  def predict_class_with_probabilities(text, model, tokenizer, max_len=128):
12
  model.eval()