Create app.py

app.py

@@ -0,0 +1,40 @@
+import streamlit as st
+import transformers
+from transformers import pipeline
+import torch
+torch.manual_seed(0)
+
+model_name = "orzhan/ruroberta-ruatd-binary"
+
+@st.cache
+def load_model(model_name):
+      model = pipeline(task="text-classification",model=model_name)
+	    return model
+	
+model = load_model(model_name)
+
+
+
+default_value = "всегда ненавидел этих ублюдочных тварей"
+examples = [default_value,
+"убила бы этих выродков и их родителей.",
+"пошли вы сука все на хуй со своим коронавирусом...",
+"Перед барином выслуживаешься холоп?",
+"просто , зарплату шакалов отрабатывают , а больше на крысенышей похожи...",
+"--- свой текст ---"]
+
+#prompts
+st.title("Демо определения сгенерированного текста")
+sent = st.selectbox("Пример", examples)
+if sent == "--- свой текст ---":
+    sent = st.text_area("Исходный текст", default_value)
+#if custom_sent == default_value:
+#    custom_sent = sent
+
+st.button('Сгенерировано или нет?')
+
+pred = model(sent)
+
+st.write("Преобразованный текст: ", pred)
+
+