Upload app.py

app.py

@@ -32,6 +32,11 @@ def load_model():
         # Load tokenizer from the same directory
         tokenizer = AutoTokenizer.from_pretrained("./lora_adapter")
         
+        # Fix padding token issue
+        if tokenizer.pad_token is None:
+            tokenizer.pad_token = tokenizer.eos_token
+            logger.info("Set pad_token to eos_token")
+        
         logger.info("LoRA model loaded successfully")
         return "LoRA model loaded successfully!"
         
@@ -42,6 +47,11 @@ def load_model():
         
         model_name = "microsoft/DialoGPT-medium"  # Closer to Phi-4 architecture
         tokenizer = AutoTokenizer.from_pretrained(model_name)
+        
+        # Fix padding token for fallback model too
+        if tokenizer.pad_token is None:
+            tokenizer.pad_token = tokenizer.eos_token
+            
         model = AutoModelForSequenceClassification.from_pretrained(
             model_name, 
             num_labels=3,