Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -18,6 +18,7 @@ def chat_with_model(message, history, system_message, max_tokens, temperature, t
|
|
| 18 |
|
| 19 |
messages = [{"role": "system", "content": system_message}]
|
| 20 |
|
|
|
|
| 21 |
for user_msg, assistant_msg in history:
|
| 22 |
if user_msg:
|
| 23 |
messages.append({"role": "user", "content": user_msg})
|
|
@@ -50,7 +51,7 @@ demo = gr.ChatInterface(
|
|
| 50 |
chat_with_model,
|
| 51 |
additional_inputs=[
|
| 52 |
gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
|
| 53 |
-
gr.Slider(minimum=1, maximum=1024, value=
|
| 54 |
gr.Slider(minimum=0.1, maximum=2.0, value=0.7, step=0.1, label="Temperature"),
|
| 55 |
gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"),
|
| 56 |
gr.Dropdown(lake_models, label="Select Model", value=lake_models[0])
|
|
|
|
| 18 |
|
| 19 |
messages = [{"role": "system", "content": system_message}]
|
| 20 |
|
| 21 |
+
# Append history messages
|
| 22 |
for user_msg, assistant_msg in history:
|
| 23 |
if user_msg:
|
| 24 |
messages.append({"role": "user", "content": user_msg})
|
|
|
|
| 51 |
chat_with_model,
|
| 52 |
additional_inputs=[
|
| 53 |
gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
|
| 54 |
+
gr.Slider(minimum=1, maximum=1024, value=1024, step=1, label="Max new tokens"),
|
| 55 |
gr.Slider(minimum=0.1, maximum=2.0, value=0.7, step=0.1, label="Temperature"),
|
| 56 |
gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"),
|
| 57 |
gr.Dropdown(lake_models, label="Select Model", value=lake_models[0])
|