DrWat/POCL

1from transformers import pipeline2import streamlit as st3# import pytest4classifier = pipeline("zero-shot-classification", model="MoritzLaurer/mDeBERTa-v3-base-mnli-xnli")5 6 7text_input = st.text_input('Введите текст')8 9def classify(text = None):10    if text == None:11        text = text_input12    candidate_labels = ["Учетная запись", "РПД", "Учебные планы", "Личный кабинет"]13    output = classifier(text, candidate_labels, multi_label=False, use_fast=False)14    return output["labels"][0], output["scores"][0]15 16submit = st.button('Отправить')17 18if submit:19   st.write(classify())20 21# def test_classify():22#     assert classify("Добрый день! Не могу войти в свой аккаунт. Что делать?") == ('Учетная запись', 0.7550224661827087)23#24#25# #Добрый день! Не могу войти в свой аккаунт. Что делать?26# #('Учетная запись', 0.7550224661827087)