Update app.py
Browse files
app.py
CHANGED
@@ -66,7 +66,7 @@ additional_inputs = [
|
|
66 |
label="Top-p (nucleus sampling)",
|
67 |
value=0.90,
|
68 |
minimum=0.0,
|
69 |
-
maximum=1,
|
70 |
step=0.05,
|
71 |
interactive=True,
|
72 |
info="Higher values sample more low-probability tokens",
|
@@ -91,7 +91,7 @@ with gr.Blocks() as demo:
|
|
91 |
chat_output = gr.Textbox(label="Chatbot Response")
|
92 |
temperature = gr.Slider(label="Temperature", minimum=0.1, maximum=1.0, value=0.7, step=0.1)
|
93 |
max_tokens = gr.Slider(label="Max Tokens", minimum=10, maximum=512, value=100, step=10)
|
94 |
-
top_p = gr.Slider(label="Top-p", minimum=0.1, maximum
|
95 |
repetition_penalty = gr.Slider(label="Repetition Penalty", minimum=1.0, maximum=2.0, value=1.2, step=0.1)
|
96 |
chat_button = gr.Button("Send")
|
97 |
chat_button.click(generate, inputs=[chat_input, temperature, max_tokens, top_p, repetition_penalty], outputs=chat_output)
|
|
|
66 |
label="Top-p (nucleus sampling)",
|
67 |
value=0.90,
|
68 |
minimum=0.0,
|
69 |
+
maximum=1.0,
|
70 |
step=0.05,
|
71 |
interactive=True,
|
72 |
info="Higher values sample more low-probability tokens",
|
|
|
91 |
chat_output = gr.Textbox(label="Chatbot Response")
|
92 |
temperature = gr.Slider(label="Temperature", minimum=0.1, maximum=1.0, value=0.7, step=0.1)
|
93 |
max_tokens = gr.Slider(label="Max Tokens", minimum=10, maximum=512, value=100, step=10)
|
94 |
+
top_p = gr.Slider(label="Top-p", minimum=0.1, maximum=1.0, value=0.9, step=0.1)
|
95 |
repetition_penalty = gr.Slider(label="Repetition Penalty", minimum=1.0, maximum=2.0, value=1.2, step=0.1)
|
96 |
chat_button = gr.Button("Send")
|
97 |
chat_button.click(generate, inputs=[chat_input, temperature, max_tokens, top_p, repetition_penalty], outputs=chat_output)
|