Update app.py

app.py

@@ -54,7 +54,7 @@ def summarize(video_id):
     inputs = tokenizer(texts, return_tensors="pt", padding=True, )
 
     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(input_ids=inputs["input_ids"].to(device), 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