File size: 626 Bytes
b935e3d f404d24 b935e3d dee82dd b935e3d ed0ae5f dee82dd f404d24 dee82dd ed0ae5f f404d24 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
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()
|