Upload transformer.py

transformer.py

@@ -0,0 +1,29 @@
+import time
+
+import streamlit as st
+from transformers import pipeline
+
+
+@st.cache(allow_output_mutation=True)
+def get_generator():
+    return pipeline("text-generation", model="sberbank-ai/rugpt3small_based_on_gpt2")
+
+
+g = get_generator()
+
+st.markdown("**Генератор смешных и не очень текстов**")
+
+input_test = st.text_area("Введите начало текста:")
+length = st.slider("Отрегeлируйте длину текста", min_value=50, max_value=200)
+button = st.button("Сгенерировать!")
+
+if button:
+    res = g(
+        input_test,
+        do_sample=True,
+        max_length=length,
+        top_k=50,
+        top_p=0.95,
+        num_returned_sequences=1,
+    )
+    st.markdown(res[0]["generated_text"])