app.py

import gradio as gr
import os
from transformers import pipeline, set_seed


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


def Bemenet(bemenet):
    return generator(os.environ['SECRET_PROMPT'] + " " + 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()