Upload app.py

sber large, tuned parameters

app.py

@@ -1,10 +1,16 @@
 import gradio as gr
 from transformers import pipeline
 
-generator = pipeline('text-generation', model='sberbank-ai/rugpt3small_based_on_gpt2')
+generator = pipeline('text-generation', model='sberbank-ai/rugpt3large_based_on_gpt2')
 
 def generate(text, history):
-    result = generator(text, max_length=300, temperature=0.7, num_return_sequences=1, do_sample=True)
+    result = generator(
+        text,
+        do_sample=True, max_length=50, repetition_penalty=5.0,
+        top_k=5, top_p=0.95, temperature=1,
+        num_beams=None,
+        no_repeat_ngram_size=3
+    )
     return result[0]["generated_text"]
 
 demo = gr.ChatInterface(generate)