Update app.py
Browse files
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 |
# выводим результаты модели в текстовое поле, на потеху пользователю
|