Update app.py
Browse files
app.py
CHANGED
@@ -39,6 +39,7 @@ def respond(
|
|
39 |
outputs = pipeline(
|
40 |
messages,
|
41 |
max_new_tokens=max_tokens,
|
|
|
42 |
temperature=temperature,
|
43 |
top_p=top_p,
|
44 |
eos_token_id=terminators
|
@@ -56,7 +57,7 @@ with gr.Blocks() as demo:
|
|
56 |
respond,
|
57 |
additional_inputs=[
|
58 |
gr.Slider(minimum=1, maximum=2048, value=256, step=1, label="Max new tokens"),
|
59 |
-
gr.Slider(minimum=0.1, maximum=
|
60 |
gr.Slider(
|
61 |
minimum=0.1,
|
62 |
maximum=1.0,
|
|
|
39 |
outputs = pipeline(
|
40 |
messages,
|
41 |
max_new_tokens=max_tokens,
|
42 |
+
do_sample = True,
|
43 |
temperature=temperature,
|
44 |
top_p=top_p,
|
45 |
eos_token_id=terminators
|
|
|
57 |
respond,
|
58 |
additional_inputs=[
|
59 |
gr.Slider(minimum=1, maximum=2048, value=256, step=1, label="Max new tokens"),
|
60 |
+
gr.Slider(minimum=0.1, maximum=1.0, value=0.7, step=0.1, label="Temperature"),
|
61 |
gr.Slider(
|
62 |
minimum=0.1,
|
63 |
maximum=1.0,
|