Update app.py

app.py

@@ -142,12 +142,16 @@ class HealthAssistant:
     def __init__(self):
         self.medical_agent = MedicalAgent()
         self.tokenizer = AutoTokenizer.from_pretrained("microsoft/phi-2")
+        # Set padding token
+        self.tokenizer.pad_token = self.tokenizer.eos_token
         self.model = AutoModelForCausalLM.from_pretrained(
             "microsoft/phi-2",
             torch_dtype=torch.float16,
             device_map="auto",
             trust_remote_code=True
         )
+        # Set model's padding token
+        self.model.config.pad_token_id = self.tokenizer.pad_token_id
         
     def process_message(self, message: str) -> str:
         # Tokenize with attention mask