metastable-void
commited on
update
Browse files
app.py
CHANGED
@@ -55,8 +55,9 @@ def generate(
|
|
55 |
t = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
|
56 |
output = my_pipeline(
|
57 |
t,
|
58 |
-
)
|
59 |
-
|
|
|
60 |
|
61 |
demo = gr.ChatInterface(
|
62 |
fn=generate,
|
|
|
55 |
t = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
|
56 |
output = my_pipeline(
|
57 |
t,
|
58 |
+
)
|
59 |
+
print(output)
|
60 |
+
yield output[-1]["generated_text"]
|
61 |
|
62 |
demo = gr.ChatInterface(
|
63 |
fn=generate,
|