metastable-void commited on
Commit
1fbb9fa
·
unverified ·
1 Parent(s): 10502ca
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -55,7 +55,7 @@ def generate(
55
  t = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
56
  output = my_pipeline(
57
  t,
58
- )[-1]["generated_text"][-1]["content"]
59
  yield output
60
 
61
  demo = gr.ChatInterface(
 
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(