Update app.py

app.py

@@ -56,7 +56,7 @@ def summarize(video_id):
 
     with torch.no_grad():
         output_tokens = model.generate(*inputs, max_new_tokens=60, do_sample=True, top_p=0.9)
-        outputs = tokenizer.batch_decode(output_tokens.detach().cpu().numpy(), skip_special_tokens=True)
+        outputs = tokenizer.batch_decode(output_tokens[0].detach().cpu().numpy(), skip_special_tokens=True)
     
     return outputs