Update handler.py

handler.py

@@ -9,10 +9,13 @@ class EndpointHandler:
         print("Loading merged model...")
         self.tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH, trust_remote_code=True)
         self.model = AutoModelForCausalLM.from_pretrained(
-            MODEL_PATH, torch_dtype=torch.float16, device_map="auto", trust_remote_code=True
+            MODEL_PATH,
+            torch_dtype=torch.float16,
+            device_map="auto",
+            trust_remote_code=True
         )
         self.model.eval()
-        print("Model loaded successfully.")
+        print("✅ Model loaded successfully.")
 
     def __call__(self, data):
         prompt = data.get("inputs", "")