fix: remove [0] subscript from model output

handler.py

@@ -1,26 +1,26 @@
-from typing import Dict, Any
 from transformers import pipeline
 from PIL import Image
 import io, base64, requests
 
 class EndpointHandler:
-    def __init__(self, model_dir: str, **kwargs: Any):
+    def __init__(self, model_dir: str = "", **kwargs):
         print("🔹 Loading Swin2SR model ...")
         self.model = pipeline("image-to-image", model="sergeipetrov/swin2SR-classical-sr-x2-64")
 
-    def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
+    def __call__(self, data):
         image_input = data.get("inputs")
         if not image_input:
             return {"error": "Missing 'inputs' field"}
 
-        # Accept either a URL or a base64 string
+        # Accept URL or base64
         if isinstance(image_input, str) and image_input.startswith("http"):
             image = Image.open(requests.get(image_input, stream=True).raw).convert("RGB")
         else:
             image_bytes = base64.b64decode(image_input)
             image = Image.open(io.BytesIO(image_bytes)).convert("RGB")
 
-        result = self.model(image)[0]
+        # ✅ FIXED: remove [0]
+        result = self.model(image)
 
         buf = io.BytesIO()
         result.save(buf, format="PNG")