tolulope commited on
Commit
6c2e718
·
verified ·
1 Parent(s): 3b8fa51

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -12
app.py CHANGED
@@ -46,19 +46,14 @@ For information on how to customize the ChatInterface, peruse the gradio docs: h
46
  """
47
  demo = gr.ChatInterface(
48
  chat_function,
 
 
49
  additional_inputs=[
50
- gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
51
- gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
52
- gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
53
- gr.Slider(
54
- minimum=0.1,
55
- maximum=1.0,
56
- value=0.95,
57
- step=0.05,
58
- label="Top-p (nucleus sampling)",
59
- ),
60
- ],
61
- )
62
 
63
 
64
  if __name__ == "__main__":
 
46
  """
47
  demo = gr.ChatInterface(
48
  chat_function,
49
+ textbox=gr.Textbox(placeholder="Enter message here", container=False, scale = 7),
50
+ chatbot=gr.Chatbot(height=400),
51
  additional_inputs=[
52
+ gr.Textbox("You are helpful AI", label="System Prompt"),
53
+ gr.Slider(500,4000, label="Max New Tokens"),
54
+ gr.Slider(0,1, label="Temperature")
55
+ ]
56
+ )
 
 
 
 
 
 
 
57
 
58
 
59
  if __name__ == "__main__":