Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -4,6 +4,11 @@ import os
|
|
4 |
|
5 |
token = os.environ["TOKEN"]
|
6 |
model=os.environ["MODEL"]
|
7 |
-
demo = gr.load(model, src="spaces", token=token,
|
|
|
|
|
|
|
|
|
|
|
8 |
save_history=True)
|
9 |
demo.launch(show_api=False, show_error=False)
|
|
|
4 |
|
5 |
token = os.environ["TOKEN"]
|
6 |
model=os.environ["MODEL"]
|
7 |
+
demo = gr.load(model, src="spaces", token=token, additional_inputs=[
|
8 |
+
gr.Slider(1, 2048, value=512, step=1, label="Max new tokens",render=False),
|
9 |
+
gr.Slider(0.1, 4.0, value=0.9, step=0.1, label="Temperature",render=False),
|
10 |
+
gr.Slider(0.1, 1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)",render=False),
|
11 |
+
gr.Checkbox(value=True, visible=False, label="stream",render=False)
|
12 |
+
],editable=True,
|
13 |
save_history=True)
|
14 |
demo.launch(show_api=False, show_error=False)
|