Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ def generate(
|
|
11 |
temperature=0.4,
|
12 |
top_p=0.95,
|
13 |
top_k=50,
|
14 |
-
max_new_tokens=
|
15 |
):
|
16 |
pipe = load_model(model_name)
|
17 |
# Need to add additional options later.
|
@@ -48,7 +48,7 @@ g = gr.Interface(
|
|
48 |
gr.components.Slider(minimum=0, maximum=1, value=0.4, label="Temperature"),
|
49 |
gr.components.Slider(minimum=0, maximum=1, value=0.95, label="Top p"),
|
50 |
gr.components.Slider(minimum=0, maximum=100, step=1, value=50, label="Top k"),
|
51 |
-
gr.components.Slider(minimum=1, maximum=1024, step=1, value=
|
52 |
],
|
53 |
outputs=[gr.Textbox(lines=10, label="Output")],
|
54 |
title="Hugging Face Transformers Model",
|
|
|
11 |
temperature=0.4,
|
12 |
top_p=0.95,
|
13 |
top_k=50,
|
14 |
+
max_new_tokens=2000,
|
15 |
):
|
16 |
pipe = load_model(model_name)
|
17 |
# Need to add additional options later.
|
|
|
48 |
gr.components.Slider(minimum=0, maximum=1, value=0.4, label="Temperature"),
|
49 |
gr.components.Slider(minimum=0, maximum=1, value=0.95, label="Top p"),
|
50 |
gr.components.Slider(minimum=0, maximum=100, step=1, value=50, label="Top k"),
|
51 |
+
gr.components.Slider(minimum=1, maximum=1024, step=1, value=2000, label="Max tokens"),
|
52 |
],
|
53 |
outputs=[gr.Textbox(lines=10, label="Output")],
|
54 |
title="Hugging Face Transformers Model",
|