Update app.py
Browse files
app.py
CHANGED
|
@@ -44,21 +44,15 @@ def respond(
|
|
| 44 |
# Gradio ChatInterface with additional inputs
|
| 45 |
demo = gr.ChatInterface(
|
| 46 |
respond,
|
| 47 |
-
|
| 48 |
-
|
| 49 |
-
|
| 50 |
-
|
| 51 |
-
|
| 52 |
-
|
| 53 |
-
|
| 54 |
-
|
| 55 |
-
|
| 56 |
-
# maximum=1.0,
|
| 57 |
-
# value=0.6,
|
| 58 |
-
# step=0.05,
|
| 59 |
-
# label="Top-p (nucleus sampling)",
|
| 60 |
-
# ),
|
| 61 |
-
# ],
|
| 62 |
)
|
| 63 |
|
| 64 |
if __name__ == "__main__":
|
|
|
|
| 44 |
# Gradio ChatInterface with additional inputs
|
| 45 |
demo = gr.ChatInterface(
|
| 46 |
respond,
|
| 47 |
+
additional_inputs=[
|
| 48 |
+
# gr.Textbox(
|
| 49 |
+
# value="Hi, How can we be of service to you, today?",
|
| 50 |
+
# label="System message",
|
| 51 |
+
# ),
|
| 52 |
+
gr.Slider(minimum=1, maximum=512, value=170, step=1, label="Max new tokens"),
|
| 53 |
+
gr.Slider(minimum=0.1, maximum=1.0, value=0.2, step=0.1, label="Temperature"),
|
| 54 |
+
gr.Slider(minimum=0.1,maximum=1.0,value=0.8,step=0.05,label="Top-p (nucleus sampling)",),
|
| 55 |
+
],
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 56 |
)
|
| 57 |
|
| 58 |
if __name__ == "__main__":
|