Update paraphraser.py

paraphraser.py

@@ -27,12 +27,20 @@ def paraphrase_comment(comment):
             f"Provide the paraphrased comment only, without additional explanation."
         )
 
-        # Tokenize the prompt
-        inputs = tokenizer(prompt, return_tensors="pt", truncation=True, padding=True, max_length=512)
+        # Tokenize the prompt with attention mask
+        inputs = tokenizer(
+            prompt,
+            return_tensors="pt",
+            truncation=True,
+            padding=True,
+            max_length=512,
+            return_attention_mask=True
+        ).to(model.device)
 
         # Generate the paraphrased comment
         outputs = model.generate(
-            inputs["input_ids"],
+            input_ids=inputs["input_ids"],
+            attention_mask=inputs["attention_mask"],
             max_length=512,
             num_beams=5,
             no_repeat_ngram_size=2,