| import gradio as gr | |
| from transformers import pipeline | |
| # Load an available model for text generation | |
| model = pipeline("text-generation", model="gpt2") | |
| def generate_text(prompt): | |
| return model(prompt, max_length=50)[0]["generated_text"] | |
| # Custom interface with a textbox and description | |
| interface = gr.Interface( | |
| fn=generate_text, | |
| inputs=gr.Textbox(label="Enter your prompt here"), | |
| outputs=gr.Textbox(label="Generated Text"), | |
| title="AI Text Generator", | |
| description="This app generates text based on your input prompt. Try it out!", | |
| theme="dark" | |
| ) | |
| if __name__ == "__main__": | |
| interface.launch() | |