Update medical_simplifier.py

medical_simplifier.py

@@ -32,7 +32,15 @@ class MedicalTextSimplifier:
         """Generate plain-language explanation using BioMedLM"""
         try:
             prompt = f"Explain the medical term '{term}' in simple language for a patient. Context: {context}\nExplanation:"
-            result = self.pipe(prompt, max_length=100, do_sample=True, temperature=0.7, top_p=0.9)
+            result = self.pipe(
+            prompt, 
+            max_new_tokens=50,  # Use max_new_tokens instead of max_length
+            do_sample=True, 
+            temperature=0.7, 
+            top_p=0.9,
+            truncation=True,  # Explicitly enable truncation
+            pad_token_id=self.pipe.tokenizer.eos_token_id  # Set pad token
+            )
             explanation = result[0]['generated_text'].split("Explanation:")[-1].strip()
             return explanation
         except Exception as e:
@@ -87,4 +95,6 @@ class MedicalTextSimplifier:
             simplified_text = simplified_text[:start] + annotated + simplified_text[end:]
             offset += len(annotated) - len(term)
             
+        print("\nSimplified text:")
+        print(simplified_text)    
         return simplified_text