Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -81,24 +81,25 @@ with gr.Blocks(css=css) as demo:
|
|
81 |
system_input = gr.Textbox(
|
82 |
label="System Prompt",
|
83 |
value=DEFAULT_SYSTEM_PROMPT,
|
84 |
-
elem_classes="input-box"
|
|
|
85 |
)
|
86 |
-
|
87 |
-
|
88 |
-
|
89 |
-
|
90 |
-
|
91 |
-
|
92 |
-
|
93 |
-
|
94 |
-
|
95 |
-
|
96 |
-
|
97 |
-
|
98 |
-
|
99 |
-
|
100 |
-
|
101 |
-
|
102 |
generate_btn.click(
|
103 |
fn=execute_jupyter_agent,
|
104 |
inputs=[system_input, user_input, max_tokens, model, state],
|
|
|
81 |
system_input = gr.Textbox(
|
82 |
label="System Prompt",
|
83 |
value=DEFAULT_SYSTEM_PROMPT,
|
84 |
+
elem_classes="input-box",
|
85 |
+
lines=8
|
86 |
)
|
87 |
+
with gr.Row():
|
88 |
+
max_tokens = gr.Number(
|
89 |
+
label="Max New Tokens",
|
90 |
+
value=DEFAULT_MAX_TOKENS,
|
91 |
+
minimum=128,
|
92 |
+
maximum=2048,
|
93 |
+
step=8,
|
94 |
+
interactive=True
|
95 |
+
)
|
96 |
+
|
97 |
+
model = gr.Dropdown(choices=[
|
98 |
+
"meta-llama/Llama-3.2-3B-Instruct",
|
99 |
+
"meta-llama/Llama-3.1-8B-Instruct",
|
100 |
+
"meta-llama/Llama-3.1-70B-Instruct"]
|
101 |
+
)
|
102 |
+
|
103 |
generate_btn.click(
|
104 |
fn=execute_jupyter_agent,
|
105 |
inputs=[system_input, user_input, max_tokens, model, state],
|