import gradio as gr
from utils import generate

interface = gr.Interface(
    fn=generate,
    inputs=[
        gr.Textbox(label="Prompt", lines=5),
        gr.Number(label="Max Tokens", value=100)
    ],
    outputs=gr.Textbox(label="Response"),
    title="🧠 Qwen2-0.5B Raw API",
    description="Raw access to the Qwen2-0.5B model. Send your full prompt."
)

if __name__ == "__main__":
    interface.launch()