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,
|