Refactor EndpointHandler constructor to use model_dir parameter for consistency

handler.py

@@ -1,25 +1,24 @@
 import base64
 import io
-from PIL import Image
 import torch
-
+from PIL import Image
 from transformers import Qwen2_5_VLForConditionalGeneration, AutoProcessor
 from qwen_vl_utils import process_vision_info
 
 
 class EndpointHandler:
 
-    def __init__(self, path=""):
+    def __init__(self, model_dir):
 
         self.model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
-            path,
+            model_dir,
             torch_dtype=torch.bfloat16,
             device_map="auto",
             trust_remote_code=True
         )
 
         self.processor = AutoProcessor.from_pretrained(
-            path,
+            model_dir,
             trust_remote_code=True
         )
 
@@ -53,7 +52,7 @@ class EndpointHandler:
             images=image_inputs,
             videos=video_inputs,
             padding=True,
-            return_tensors="pt",
+            return_tensors="pt"
         ).to(self.model.device)
 
         outputs = self.model.generate(**inputs, max_new_tokens=512)
@@ -68,4 +67,4 @@ class EndpointHandler:
             skip_special_tokens=True
         )
 
-        return decoded[0]
+        return {"result": decoded[0]}