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)
|