BK-Lee commited on
Commit
74321a7
·
1 Parent(s): a29da54
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -125,7 +125,7 @@ def bot_streaming(message, history, link, temperature, new_max_token, top_p):
125
  yield buffer
126
 
127
  demo = gr.ChatInterface(fn=bot_streaming,
128
- additional_inputs = [gr.Radio(["1.8B", "3.8B", "7B"], label="Size", info="Select one model size"), gr.Slider(0, 1, 0.9, label="temperature"), gr.Slider(1, 1024, 128, label="new_max_token"), gr.Slider(0, 1, 0.95, label="top_p")],
129
  additional_inputs_accordion="Generation Hyperparameters",
130
  theme=gr.themes.Soft(),
131
  title="TroL",
 
125
  yield buffer
126
 
127
  demo = gr.ChatInterface(fn=bot_streaming,
128
+ additional_inputs = [gr.Radio(["1.8B", "3.8B", "7B"], label="Size", info="Select one model size", value="7B"), gr.Slider(0, 1, 0.9, label="temperature"), gr.Slider(1, 1024, 128, label="new_max_token"), gr.Slider(0, 1, 0.95, label="top_p")],
129
  additional_inputs_accordion="Generation Hyperparameters",
130
  theme=gr.themes.Soft(),
131
  title="TroL",