Update paraphraser.py

paraphraser.py

@@ -42,6 +42,9 @@ def paraphrase_comment(comment):
         paraphrased_comment = tokenizer.decode(outputs[0], skip_special_tokens=True)
         # Remove the prompt part from the output
         paraphrased_comment = paraphrased_comment.replace(prompt, "").strip()
+        # Remove unwanted prefixes like "Neutral: "
+        if paraphrased_comment.startswith("Neutral: "):
+            paraphrased_comment = paraphrased_comment[len("Neutral: "):].strip()
         return paraphrased_comment
 
     except Exception as e: