Dzhamb/ArticleClassification

1import streamlit as st2import torch3from transformers import DistilBertForSequenceClassification, DistilBertTokenizerFast4from utils import get_text, get_labels, load_model5 6count_labels = 87 8st.markdown("## Классификатор статей")9st.markdown("Сервис классифицирует статьи по названию и аннотации. Нужно ввести в каждое окошко свою сущность и вам выдадут к какому классу относится статья")10 11 12title = st.text_area("Введите название статьи")13 14abstract = st.text_area("Введите аннотацию к статье, abstract статьи")15if st.button("Классифицируй!"):16  model, tokenizer = load_model()17  #model.load_state_dict(torch.load('weight_model'))18  19  text = get_text(title, abstract)20  if text:21    raw_predictions = get_labels(text, model, tokenizer)22    st.markdown("Список классов к которым может относится данная статья")23    for raw in raw_predictions:24      st.markdown(f"{raw}")25  else:26    st.markdown("Ваш запрос пуст. Введите хотя бы название")27 28 29