import gradio as gr
from transformers import pipeline

pipe = pipeline("text-generation", model="your-username/your-model")

def generate_text(prompt):
    return pipe(prompt, max_length=50)[0]['generated_text']

gr.Interface(fn=generate_text, inputs="text", outputs="text").launch()