Hugging Face's logo

Spaces:
Vladislawoo
/
nlp-gpt-team
Sleeping

App Files Community
2
Vladislawoo commited on
Commit
b007c9c
·
1 Parent(s): 9302909

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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('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)
 
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)