metastable-void commited on
Commit
9d7e24a
·
unverified ·
1 Parent(s): 1fbb9fa
Files changed (1) hide show
  1. app.py +3 -2
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
- )[-1]
59
- yield output
 
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,