Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -81,14 +81,7 @@ with gr.Blocks() as demo:
|
|
81 |
interactive=True,
|
82 |
label="Max New Tokens",
|
83 |
)
|
84 |
-
|
85 |
-
minimum=0.05,
|
86 |
-
maximum=1.0,
|
87 |
-
value=0.95,
|
88 |
-
step=0.05,
|
89 |
-
interactive=True,
|
90 |
-
label="Top-p (nucleus sampling)",
|
91 |
-
)
|
92 |
top_k = gr.Slider(
|
93 |
minimum=1,
|
94 |
maximum=50,
|
@@ -108,7 +101,7 @@ with gr.Blocks() as demo:
|
|
108 |
|
109 |
user_text.submit(
|
110 |
run_generation,
|
111 |
-
[user_text,
|
112 |
model_output,
|
113 |
)
|
114 |
button_submit.click(
|
|
|
81 |
interactive=True,
|
82 |
label="Max New Tokens",
|
83 |
)
|
84 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
85 |
top_k = gr.Slider(
|
86 |
minimum=1,
|
87 |
maximum=50,
|
|
|
101 |
|
102 |
user_text.submit(
|
103 |
run_generation,
|
104 |
+
[user_text,repetition_penalty=1.5,temperature,do_sample=True,top_k, max_new_tokens],
|
105 |
model_output,
|
106 |
)
|
107 |
button_submit.click(
|