import gradio as gr
from transformers import pipeline

# Load a model (use a suitable model available on Hugging Face)
model_name = "gpt2"  # Change to your desired model
generator = pipeline("text-generation", model=model_name)

# Inference function
def generate_text(prompt):
    return generator(prompt, max_length=50)[0]['generated_text']

# Gradio interface
interface = gr.Interface(fn=generate_text, 
                         inputs="text", 
                         outputs="text", 
                         title="Text Generation with GPT-2",
                         description="Enter a prompt to generate text.")

# Launch the interface
interface.launch()