v1
Browse files
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",
|