Update app.py

app.py

@@ -61,6 +61,11 @@ def load_model():
         print(f"Loading model {MODEL_NAME}...")
         model = AutoModelForCausalLM.from_pretrained(MODEL_NAME)
         tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
+
+        if tokenizer.pad_token is None:
+            tokenizer.pad_token = tokenizer.eos_token
+
+        
         print(f"Model {MODEL_NAME} loaded successfully.")
         return model, tokenizer
     except Exception as e: