Upload app.py

Answer generation via ruGPT-3 small

app.py

@@ -1,10 +1,16 @@
 import gradio as gr
-import random
+from transformers import pipeline
 
-def predict(message, history):
-    return random.choice(["Гаф гаф!", "Гаф"])
+generator = pipeline('text-generation', model='sberbank-ai/rugpt3small_based_on_gpt2')
 
-interface = gr.ChatInterface(predict)
+def generate(text):
+    result = generator(text, max_length=300, temperature=0.7, num_return_sequences=1, do_sample=True)
+    return result[0]["generated_text"]
 
-if __name__ == '__main__':
-    interface.launch(predict)
+demo = gr.Interface(
+    fn=generate,
+    inputs = gr.inputs.Textbox(lines=3, label="Input Text"),
+    outputs=gr.outputs.Textbox(label="Generated Text"),
+)
+
+demo.launch()