import gradio as gr
from transformers import pipeline

generator = pipeline('text-generation', model='sberbank-ai/rugpt3large_based_on_gpt2')

def generate(text, history):
    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)

demo.launch(share=True)