Vladislawoo commited on
Commit
f70ad97
·
1 Parent(s): b007c9c
app.py CHANGED
@@ -14,7 +14,7 @@ scaler = load('scaler.joblib')
14
 
15
  tukinazor = load('tokenizer.pkl')
16
  rnn_model = load_model('path_to_my_model.h5')
17
- bert_model = BertForSequenceClassification.from_pretrained('Vladislawoo/nlp-gpt-team')
18
  tokenizer = BertTokenizer.from_pretrained('bert-base-multilingual-cased')
19
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
20
  bert_model = bert_model.to(device)
 
14
 
15
  tukinazor = load('tokenizer.pkl')
16
  rnn_model = load_model('path_to_my_model.h5')
17
+ bert_model = BertForSequenceClassification.from_pretrained('my_bert_model')
18
  tokenizer = BertTokenizer.from_pretrained('bert-base-multilingual-cased')
19
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
20
  bert_model = bert_model.to(device)
config.json → my_bert_model/config.json RENAMED
File without changes
pytorch_model.bin → my_bert_model/pytorch_model.bin RENAMED
File without changes
special_tokens_map.json → my_bert_model/special_tokens_map.json RENAMED
File without changes
tokenizer_config.json → my_bert_model/tokenizer_config.json RENAMED
File without changes
vocab.txt → my_bert_model/vocab.txt RENAMED
File without changes