Arthur Samuel Galego Panucci FIgueiredo
commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -56,7 +56,7 @@ def chat(user_input, system_prompt, temperature, top_p, max_tokens):
|
|
| 56 |
do_sample=True,
|
| 57 |
temperature=float(temperature),
|
| 58 |
top_p=float(top_p),
|
| 59 |
-
repetition_penalty=1.
|
| 60 |
eos_token_id=tokenizer.eos_token_id,
|
| 61 |
pad_token_id=tokenizer.pad_token_id
|
| 62 |
)
|
|
@@ -119,15 +119,15 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
|
|
| 119 |
)
|
| 120 |
|
| 121 |
temperature = gr.Slider(
|
| 122 |
-
0.1, 1.5, value=0.
|
| 123 |
)
|
| 124 |
|
| 125 |
top_p = gr.Slider(
|
| 126 |
-
0.1, 1.0, value=0.
|
| 127 |
)
|
| 128 |
|
| 129 |
max_tokens = gr.Slider(
|
| 130 |
-
50,
|
| 131 |
)
|
| 132 |
|
| 133 |
submit.click(
|
|
|
|
| 56 |
do_sample=True,
|
| 57 |
temperature=float(temperature),
|
| 58 |
top_p=float(top_p),
|
| 59 |
+
repetition_penalty=1.2,
|
| 60 |
eos_token_id=tokenizer.eos_token_id,
|
| 61 |
pad_token_id=tokenizer.pad_token_id
|
| 62 |
)
|
|
|
|
| 119 |
)
|
| 120 |
|
| 121 |
temperature = gr.Slider(
|
| 122 |
+
0.1, 1.5, value=0.65, step=0.05, label="Temperature"
|
| 123 |
)
|
| 124 |
|
| 125 |
top_p = gr.Slider(
|
| 126 |
+
0.1, 1.0, value=0.95, step=0.05, label="Top-p"
|
| 127 |
)
|
| 128 |
|
| 129 |
max_tokens = gr.Slider(
|
| 130 |
+
50, 1024, value=200, step=10, label="Max tokens"
|
| 131 |
)
|
| 132 |
|
| 133 |
submit.click(
|