import gradio as gr

gr.load("models/CreitinGameplays/bloom-3b-conversational").launch()

interface = gr.Interface(
  fn=lambda prompt="", max_tokens=128: gr.Textbox.submit(value=prompt),
  inputs=[
      gr.Textbox(label="Text Prompt (optional)", value="Hi! How can I help you today?"),
      gr.Slider(minimum=1, maximum=1024, label="Max New Tokens", value=128),
  ],
  outputs=[],
  title="Bloom 3b Conversational Assistant",
  description="Talk to Bloom 3b using a text prompt and adjust the maximum number of tokens for response generation.",
)

interface.launch()