import gradio as gr
from transformers import pipeline

pipeline = pipeline(task="text-generation", model="gpt2")

def greet(name):
    return "Hello " + name + "!!"

def gen_text(text):
    if len(text) > 0:
        return pipeline(text)[0]
    else:
        return "Enter text"
    
iface = gr.Interface(fn=gen_text, 
                     inputs="text", 
                     outputs="text",
                     title="Text Generation using GPT-2",
                     description="Example of text generation using OpenAI GPT-2",
                    )
iface.launch()