import gradio as gr
from gemma_model import generate_text

chatbot = gr.Chatbot(
    generate_text,
    title="Gemma-2B-IT Chatbot",
    description="Talk to the Gemma-2B-IT model!",
    width=800,
    height=600,
)

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