charanhu commited on
Commit
5995eff
·
1 Parent(s): 3ac57f8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -6
app.py CHANGED
@@ -45,12 +45,16 @@ def predict(message, history, temperature, max_new_tokens, min_new_tokens):
45
  yield partial_message
46
 
47
 
48
- iface = gr.ChatInterface(
49
  fn=predict,
50
- inputs=["text", "text", gr.Slider(minimum=0.1, maximum=2.0, value=1.0, label="Temperature"),
51
- gr.Slider(minimum=1, maximum=2048, value=1024, label="Max Tokens"),
52
- gr.Slider(minimum=1, maximum=1024, value=1, label="Min Tokens")],
53
- outputs="text"
 
 
 
 
54
  )
55
 
56
- iface.queue().launch()
 
45
  yield partial_message
46
 
47
 
48
+ iface = gr.Interface(
49
  fn=predict,
50
+ inputs=["text", "text", gr.Slider(minimum=0.1, maximum=2.0, default=1.0, label="Temperature"),
51
+ gr.Slider(minimum=1, maximum=2048, default=1024, label="Max Tokens"),
52
+ gr.Slider(minimum=1, maximum=1024, default=1, label="Min Tokens")],
53
+ outputs="text",
54
+ live=True,
55
+ capture_session=True,
56
+ layout="vertical",
57
+ chat=True
58
  )
59
 
60
+ iface.launch()