Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -61,24 +61,24 @@ with gr.Blocks() as demo:
|
|
61 |
lines=6)
|
62 |
max_new_tokens = gr.Slider(
|
63 |
label='Max new tokens',
|
64 |
-
minimum=
|
65 |
maximum=1024,
|
66 |
step=1,
|
67 |
value=512,
|
68 |
)
|
69 |
temperature = gr.Slider(
|
70 |
label='Temperature',
|
71 |
-
minimum=0.
|
72 |
maximum=0.5,
|
73 |
-
step=0.
|
74 |
-
value=0.
|
75 |
)
|
76 |
top_p = gr.Slider(
|
77 |
label='Top-p (nucleus sampling)',
|
78 |
-
minimum=0.
|
79 |
maximum=1.0,
|
80 |
-
step=0.
|
81 |
-
value=0.
|
82 |
)
|
83 |
|
84 |
|
@@ -88,6 +88,9 @@ with gr.Blocks() as demo:
|
|
88 |
|
89 |
def bot(history, max_new_tokens, temperature, top_p, system_prompt):
|
90 |
chat_data = []
|
|
|
|
|
|
|
91 |
for user_msg, assistant_msg in history:
|
92 |
if user_msg is not None:
|
93 |
chat_data.append({"role": "user", "content": user_msg})
|
|
|
61 |
lines=6)
|
62 |
max_new_tokens = gr.Slider(
|
63 |
label='Max new tokens',
|
64 |
+
minimum=32,
|
65 |
maximum=1024,
|
66 |
step=1,
|
67 |
value=512,
|
68 |
)
|
69 |
temperature = gr.Slider(
|
70 |
label='Temperature',
|
71 |
+
minimum=0.01,
|
72 |
maximum=0.5,
|
73 |
+
step=0.01,
|
74 |
+
value=0.01,
|
75 |
)
|
76 |
top_p = gr.Slider(
|
77 |
label='Top-p (nucleus sampling)',
|
78 |
+
minimum=0.01,
|
79 |
maximum=1.0,
|
80 |
+
step=0.01,
|
81 |
+
value=0.01,
|
82 |
)
|
83 |
|
84 |
|
|
|
88 |
|
89 |
def bot(history, max_new_tokens, temperature, top_p, system_prompt):
|
90 |
chat_data = []
|
91 |
+
system_prompt = system_prompt.strip()
|
92 |
+
if system_prompt:
|
93 |
+
chat_data.append({"role": "system", "content": system_prompt})
|
94 |
for user_msg, assistant_msg in history:
|
95 |
if user_msg is not None:
|
96 |
chat_data.append({"role": "user", "content": user_msg})
|