BK-Lee commited on
Commit
eb8fafa
1 Parent(s): a45c9b1
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -138,7 +138,7 @@ def bot_streaming(message, history, link, temperature, new_max_token, top_p):
138
  yield buffer
139
 
140
  demo = gr.ChatInterface(fn=bot_streaming,
141
- additional_inputs = [gr.Radio(["1.8B", "3.8B", "7B"], label="Size", info="Select one model size", value="3.8B"), 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")],
142
  additional_inputs_accordion="Generation Hyperparameters",
143
  theme=gr.themes.Soft(),
144
  title="TroL",
 
138
  yield buffer
139
 
140
  demo = gr.ChatInterface(fn=bot_streaming,
141
+ additional_inputs = [gr.Radio(["3.8B"], label="Size", info="Select one model size", value="3.8B"), 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")],
142
  additional_inputs_accordion="Generation Hyperparameters",
143
  theme=gr.themes.Soft(),
144
  title="TroL",