Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -45,12 +45,16 @@ def predict(message, history, temperature, max_new_tokens, min_new_tokens):
|
|
45 |
yield partial_message
|
46 |
|
47 |
|
48 |
-
iface = gr.
|
49 |
fn=predict,
|
50 |
-
inputs=["text", "text", gr.Slider(minimum=0.1, maximum=2.0,
|
51 |
-
gr.Slider(minimum=1, maximum=2048,
|
52 |
-
gr.Slider(minimum=1, maximum=1024,
|
53 |
-
outputs="text"
|
|
|
|
|
|
|
|
|
54 |
)
|
55 |
|
56 |
-
iface.
|
|
|
45 |
yield partial_message
|
46 |
|
47 |
|
48 |
+
iface = gr.Interface(
|
49 |
fn=predict,
|
50 |
+
inputs=["text", "text", gr.Slider(minimum=0.1, maximum=2.0, default=1.0, label="Temperature"),
|
51 |
+
gr.Slider(minimum=1, maximum=2048, default=1024, label="Max Tokens"),
|
52 |
+
gr.Slider(minimum=1, maximum=1024, default=1, label="Min Tokens")],
|
53 |
+
outputs="text",
|
54 |
+
live=True,
|
55 |
+
capture_session=True,
|
56 |
+
layout="vertical",
|
57 |
+
chat=True
|
58 |
)
|
59 |
|
60 |
+
iface.launch()
|