Spaces:
Runtime error
Runtime error
Matt
commited on
Commit
•
9512d4d
1
Parent(s):
93451f9
Update
Browse files
app.py
CHANGED
@@ -17,12 +17,13 @@ conversations = [demo_conversation1, demo_conversation2]
|
|
17 |
|
18 |
def apply_chat_template(template):
|
19 |
tokenizer.chat_template = template
|
20 |
-
|
21 |
for i, conversation in enumerate(conversations):
|
22 |
without_gen = tokenizer.apply_chat_template(conversation, tokenize=False)
|
23 |
with_gen = tokenizer.apply_chat_template(conversation, tokenize=False, add_generation_prompt=True)
|
24 |
-
out
|
25 |
-
|
|
|
26 |
|
27 |
-
iface = gr.Interface(fn=apply_chat_template, inputs="text", outputs="text")
|
28 |
iface.launch()
|
|
|
17 |
|
18 |
def apply_chat_template(template):
|
19 |
tokenizer.chat_template = template
|
20 |
+
outputs = []
|
21 |
for i, conversation in enumerate(conversations):
|
22 |
without_gen = tokenizer.apply_chat_template(conversation, tokenize=False)
|
23 |
with_gen = tokenizer.apply_chat_template(conversation, tokenize=False, add_generation_prompt=True)
|
24 |
+
out = f"Conversation {i} without generation prompt:\n\n{without_gen}\n\nConversation {i} with generation prompt:\n\n{with_gen}\n\n"
|
25 |
+
outputs.append(out)
|
26 |
+
return *outputs
|
27 |
|
28 |
+
iface = gr.Interface(fn=apply_chat_template, inputs="text", outputs=["text"] * len(conversations))
|
29 |
iface.launch()
|