Update app.py

app.py

@@ -52,9 +52,10 @@ def summarize(video_id):
 
     texts = preprocessing(transcript)
     inputs = tokenizer(texts, return_tensors="pt", padding=True, )
+    inputs = inputs["input_ids"].to(device)
 
     with torch.no_grad():
-        output_tokens = model.generate(input_ids=inputs["input_ids"].to(device), max_new_tokens=60, do_sample=True, top_p=0.9)
+        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)
     
     return outputs