import gradio as gr
from transformers import pipeline

pipe = pipeline(
    "text-generation",
    model="ibm-granite/granite-3.3-2b-base"
)

def generate_text(prompt, max_new_tokens=100):
    result = pipe(prompt, max_new_tokens=max_new_tokens, do_sample=True)
    return result[0]["generated_text"]

interface = gr.Interface(
    fn=generate_text,
    inputs=[
        gr.Textbox(lines=5, placeholder="Enter your prompt here..."),
        gr.Slider(10, 300, value=100, step=10)
    ],
    outputs="text",
    title="Granite Text Generator"
)

interface.launch()