Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ from threading import Thread
|
|
9 |
HF_TOKEN = os.environ.get("HF_TOKEN", None)
|
10 |
MODEL = "AGI-0/Art-Skynet-3B"
|
11 |
|
12 |
-
TITLE = """<h2>Link to the model: <a href="https://huggingface.co/AGI-0/Art-
|
13 |
PLACEHOLDER = """<center><p>Hi! How can I help you today?</p></center>"""
|
14 |
CSS = """.duplicate-button { margin: auto !important; color: white !important; background: black !important; border-radius: 100vh !important; } h3 { text-align: center; }"""
|
15 |
|
@@ -196,10 +196,10 @@ with gr.Blocks(css=CSS, theme="soft") as demo:
|
|
196 |
),
|
197 |
additional_inputs=[
|
198 |
gr.Textbox(value="", label="System Prompt", render=False),
|
199 |
-
gr.Slider(minimum=0, maximum=1, step=0.1, value=0.
|
200 |
gr.Slider(minimum=128, maximum=8192, step=1, value=4096, label="Max new tokens", render=False),
|
201 |
gr.Slider(minimum=0.0, maximum=1.0, step=0.1, value=1.0, label="top_p", render=False),
|
202 |
-
gr.Slider(minimum=1, maximum=50, step=1, value=
|
203 |
gr.Slider(minimum=0.0, maximum=2.0, step=0.1, value=1.1, label="Repetition penalty", render=False),
|
204 |
],
|
205 |
examples=[
|
|
|
9 |
HF_TOKEN = os.environ.get("HF_TOKEN", None)
|
10 |
MODEL = "AGI-0/Art-Skynet-3B"
|
11 |
|
12 |
+
TITLE = """<h2>Link to the model: <a href="https://huggingface.co/AGI-0/Art-Skynet-3B">click here</a>(please leave a like on the repo)</h2>"""
|
13 |
PLACEHOLDER = """<center><p>Hi! How can I help you today?</p></center>"""
|
14 |
CSS = """.duplicate-button { margin: auto !important; color: white !important; background: black !important; border-radius: 100vh !important; } h3 { text-align: center; }"""
|
15 |
|
|
|
196 |
),
|
197 |
additional_inputs=[
|
198 |
gr.Textbox(value="", label="System Prompt", render=False),
|
199 |
+
gr.Slider(minimum=0, maximum=1, step=0.1, value=0.6, label="Temperature", render=False),
|
200 |
gr.Slider(minimum=128, maximum=8192, step=1, value=4096, label="Max new tokens", render=False),
|
201 |
gr.Slider(minimum=0.0, maximum=1.0, step=0.1, value=1.0, label="top_p", render=False),
|
202 |
+
gr.Slider(minimum=1, maximum=50, step=1, value=25, label="top_k", render=False),
|
203 |
gr.Slider(minimum=0.0, maximum=2.0, step=0.1, value=1.1, label="Repetition penalty", render=False),
|
204 |
],
|
205 |
examples=[
|