import gradio as gr
from transformers import pipeline, set_seed


generator = pipeline('text-generation', model='gpt2', return_full_text=False)
set_seed(42)


def Bemenet(bemenet):
    return generator(bemenet, max_length=100, num_return_sequences=2)[1]['generated_text']


interface = gr.Interface(fn=Bemenet,
                        title="Cím..",
                        description="Leírás..",
                        inputs="text",
                        outputs="text")

interface.launch()