import gradio as gr
from transformers import pipeline

generator = pipeline("text-generation", model = "distilgpt2")


def generate_text(prompt):
    generated_text = generator(prompt, max_length=30, num_return_sequences=1)[0]['generated_text']
    return generated_text

iface = gr.Interface(fn=generate_text, inputs="text", 
                     outputs="text",
                     live=True,
                     title="Text Generator App 📝📝📝📝",
                     description="Enter a text.")
iface.launch(share=True)