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
@@ -65,4 +66,4 @@ gr.Interface(
     # inputs = gr.File(file_types=["text"], label="Upload a text file.", interactive=True),
     inputs = gr.Textbox(label="Video_ID", interactive=True),
     outputs = gr.Textbox(label="Summary", max_lines=120, interactive=False),
-).launch(debug=True)
+).launch(debug=True)