Spaces:
Runtime error
Runtime error
Matt
commited on
Commit
·
69460b6
1
Parent(s):
3396f08
Update
Browse files
app.py
CHANGED
@@ -13,6 +13,13 @@ demo_conversation2 = [
|
|
13 |
{"role": "user", "content": "Hi there!"}
|
14 |
]
|
15 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
16 |
conversations = [demo_conversation1, demo_conversation2]
|
17 |
|
18 |
def apply_chat_template(template):
|
@@ -25,5 +32,8 @@ def apply_chat_template(template):
|
|
25 |
outputs.append(out)
|
26 |
return tuple(outputs)
|
27 |
|
28 |
-
iface = gr.Interface(
|
|
|
|
|
|
|
29 |
iface.launch()
|
|
|
13 |
{"role": "user", "content": "Hi there!"}
|
14 |
]
|
15 |
|
16 |
+
default_template = """{% for message in messages %}
|
17 |
+
{{ "<|im_start|>" + message["role"] + \n + message["content"] + "<|im_end|>\n" }}
|
18 |
+
{% endfor %}
|
19 |
+
{% if add_generation_prompt %}
|
20 |
+
{{ "<|im_start|>assistant\n" }}
|
21 |
+
{% endif %}"""
|
22 |
+
|
23 |
conversations = [demo_conversation1, demo_conversation2]
|
24 |
|
25 |
def apply_chat_template(template):
|
|
|
32 |
outputs.append(out)
|
33 |
return tuple(outputs)
|
34 |
|
35 |
+
iface = gr.Interface(
|
36 |
+
fn=apply_chat_template,
|
37 |
+
inputs=gr.TextArea(value=default_template, lines=10, max_lines=30, label="Chat Template"),
|
38 |
+
outputs=["text"] * len(conversations))
|
39 |
iface.launch()
|