Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -68,12 +68,27 @@ def generate(
|
|
68 |
demo = gr.ChatInterface(
|
69 |
fn=generate,
|
70 |
additional_inputs=[
|
71 |
-
gr.Slider(
|
72 |
-
|
73 |
-
|
74 |
-
|
75 |
-
|
76 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
77 |
],
|
78 |
stop_btn=None,
|
79 |
examples=[["Write me an English pangram."]],
|
|
|
68 |
demo = gr.ChatInterface(
|
69 |
fn=generate,
|
70 |
additional_inputs=[
|
71 |
+
gr.Slider(
|
72 |
+
label="Max new tokens",
|
73 |
+
minimum=1,
|
74 |
+
maximum=MAX_MAX_NEW_TOKENS,
|
75 |
+
step=1,
|
76 |
+
value=DEFAULT_MAX_NEW_TOKENS,
|
77 |
+
),
|
78 |
+
gr.Slider(
|
79 |
+
label="Temperature",
|
80 |
+
minimum=0.1,
|
81 |
+
maximum=4.0,
|
82 |
+
step=0.1,
|
83 |
+
value=0.6,
|
84 |
+
),
|
85 |
+
gr.Slider(
|
86 |
+
label="Top-p (nucleus sampling)",
|
87 |
+
minimum=0.05,
|
88 |
+
maximum=1.0,
|
89 |
+
step=0.05,
|
90 |
+
value=0.9,
|
91 |
+
),
|
92 |
],
|
93 |
stop_btn=None,
|
94 |
examples=[["Write me an English pangram."]],
|