Update handler.py

handler.py

@@ -10,14 +10,21 @@ class EndpointHandler:
         self.model = CLIPModel.from_pretrained("dazpye/clip-image")
         self.processor = CLIPProcessor.from_pretrained("dazpye/clip-image")
 
-    def _load_image(self, image_data):
-        """Handles both URL-based and base64 image inputs."""
-        if isinstance(image_data, str):  
-            if image_data.startswith("http"):  
-                return Image.open(requests.get(image_data, stream=True).raw)
-            else:  # Assume base64-encoded image
-                return Image.open(io.BytesIO(base64.b64decode(image_data)))
-        return None  # Invalid image format
+    def _load_image(self, image_url):
+        """Fetches an image and ensures it is fully loaded."""
+        try:
+            print(f"Fetching image from: {image_url}")
+            response = requests.get(image_url, timeout=5)
+            print(f"HTTP Status Code: {response.status_code}")
+            
+            if response.status_code == 200:
+                image_bytes = io.BytesIO(response.content)  # Convert to bytes
+                return Image.open(image_bytes)
+            else:
+                print(f"❌ Failed to fetch image: HTTP {response.status_code}")
+        except Exception as e:
+            print(f"❌ Exception in image loading: {e}")
+        return None  # Return None if image loading fails
 
     def __call__(self, data):
         """Main inference function Hugging Face expects."""