app.py

import gradio as gr
from transformers import pipeline

generator = pipeline("text-generation", "openai-community/gpt2")

def generate(text):
    output = generator(text)
    return output[0]['generated_text']

iface = gr.Interface(fn=generate, inputs="text", outputs="text", title="FlareNew Text Generation")
iface.launch()