testsson / app.py
cakemus's picture
style
ed0ae5f
raw
history blame
604 Bytes
import gradio as gr
from transformers import pipeline
# Load a large model (e.g., text generation)
model = pipeline("text-generation", model="gpt-3")
def generate_text(prompt):
return model(prompt, max_length=50)
# Custom interface with sliders and descriptions
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()