CreitinGameplays commited on
Commit
49d7b84
1 Parent(s): 5b60a87

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -1
app.py CHANGED
@@ -1,3 +1,16 @@
1
  import gradio as gr
2
 
3
- gr.load("models/CreitinGameplays/bloom-3b-conversational").launch()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import gradio as gr
2
 
3
+ gr.load("models/CreitinGameplays/bloom-3b-conversational").launch()
4
+
5
+ interface = gr.Interface(
6
+ fn=lambda prompt="", max_tokens=128: gr.Textbox.submit(value=prompt),
7
+ inputs=[
8
+ gr.Textbox(label="Text Prompt (optional)", value="Hi! How can I help you today?"),
9
+ gr.Slider(minimum=1, maximum=1024, label="Max New Tokens", value=128),
10
+ ],
11
+ outputs=[],
12
+ title="Bloom 3b Conversational Assistant",
13
+ description="Talk to Bloom 3b using a text prompt and adjust the maximum number of tokens for response generation.",
14
+ )
15
+
16
+ interface.launch()