Add application file

main.py

@@ -0,0 +1,26 @@
+from transformers import pipeline
+import streamlit as st
+# import pytest
+classifier = pipeline("zero-shot-classification", model="MoritzLaurer/mDeBERTa-v3-base-mnli-xnli")
+
+
+text_input = st.text_input('Введите текст')
+
+def classify(text = None):
+    if text == None:
+        text = text_input
+    candidate_labels = ["Учетная запись", "РПД", "Учебные планы", "Личный кабинет"]
+    output = classifier(text, candidate_labels, multi_label=False, use_fast=False)
+    return output["labels"][0], output["scores"][0]
+
+submit = st.button('Отправить')
+
+if submit:
+   st.write(classify())
+
+# def test_classify():
+#     assert classify("Добрый день! Не могу войти в свой аккаунт. Что делать?") == ('Учетная запись', 0.7550224661827087)
+#
+#
+# #Добрый день! Не могу войти в свой аккаунт. Что делать?
+# #('Учетная запись', 0.7550224661827087)

test_sample.py

@@ -0,0 +1,26 @@
+from transformers import pipeline
+import streamlit as st
+import pytest
+classifier = pipeline("zero-shot-classification", model="MoritzLaurer/mDeBERTa-v3-base-mnli-xnli")
+
+
+text_input = st.text_input('Введите текст')
+
+def classify(text = None):
+    if text == None:
+        text = text_input
+    candidate_labels = ["Учетная запись", "РПД", "Учебные планы", "Личный кабинет"]
+    output = classifier(text, candidate_labels, multi_label=False, use_fast=False)
+    return output["labels"][0], output["scores"][0]
+
+submit = st.button('Отправить')
+
+if submit:
+   st.write(classify())
+
+def test_classify():
+    assert classify("Добрый день! Не могу войти в свой аккаунт. Что делать?") == ('Учетная запись', 0.7550224661827087)
+
+
+# #Добрый день! Не могу войти в свой аккаунт. Что делать?
+# #('Учетная запись', 0.7550224661827087)