TH9817 commited on
Commit
b163c9a
·
verified ·
1 Parent(s): 36a91f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -94,7 +94,7 @@ def chat(video_number,token):
94
  output = model.generate(**inputs, **generate_kwargs)
95
  generated_text = processor.batch_decode(output, skip_special_tokens=True)
96
 
97
- return generated_text
98
 
99
  demo = gr.Interface(
100
  fn=chat,
 
94
  output = model.generate(**inputs, **generate_kwargs)
95
  generated_text = processor.batch_decode(output, skip_special_tokens=True)
96
 
97
+ return generated_text[0][45:]
98
 
99
  demo = gr.Interface(
100
  fn=chat,