freeCS-dot-org commited on
Commit
03ad009
·
verified ·
1 Parent(s): 845aeaa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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-v0-3B">click here</a></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,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.8, 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=50, 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=[
 
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=[