arkmartov commited on
Commit
7d57f72
1 Parent(s): 00e7fa3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -13,4 +13,10 @@ raw_predictions = pipe(text)
13
  # тут уже знакомый вам код с huggingface.transformers -- его можно заменить на что угодно от fairseq до catboost
14
 
15
  st.markdown(f"{raw_predictions}")
 
 
 
 
 
 
16
  # выводим результаты модели в текстовое поле, на потеху пользователю
 
13
  # тут уже знакомый вам код с huggingface.transformers -- его можно заменить на что угодно от fairseq до catboost
14
 
15
  st.markdown(f"{raw_predictions}")
16
+
17
+ from transformers import BertTokenizer, BertForSequenceClassification
18
+ model_name = "google/bert_uncased_L-4_H-256_A-4"
19
+ tokenizer = BertTokenizer.from_pretrained(model_name)
20
+ model = BertForSequenceClassification.from_pretrained(model_name, num_labels=8)
21
+ st.martkdown(f"{model}")
22
  # выводим результаты модели в текстовое поле, на потеху пользователю