Update handler.py

handler.py

@@ -34,6 +34,13 @@ class EndpointHandler():
         if not image_url or not prompt:
             return {"error": "Both 'url' and 'prompt' must be provided in 'inputs'."}
         
+        # Depuració
+        debug_info = {
+            "stage": "processing input",
+            "image_url": image_url,
+            "prompt": prompt
+        }
+
         try:
             response = requests.get(image_url, stream=True)
             image = Image.open(response.raw)
@@ -46,15 +53,23 @@ class EndpointHandler():
                 image = Image.open(buffer)
 
         except Exception as e:
-            return {"error": str(e)}
+            debug_info["stage"] = "loading image"
+            debug_info["error"] = str(e)
+            return debug_info
 
-        inputs = self.processor(prompt, image, return_tensors="pt").to("cuda")
-        output = self.model.generate(**inputs, max_new_tokens=100)
-        result = self.processor.decode(output[0], skip_special_tokens=True)
-        
-        scores = self.extract_scores(result)
-        sorted_scores = sorted(scores.items(), key=lambda item: item[1], reverse=True)
-        return sorted_scores
+        try:
+            inputs = self.processor(prompt, image, return_tensors="pt").to("cuda")
+            output = self.model.generate(**inputs, max_new_tokens=100)
+            result = self.processor.decode(output[0], skip_special_tokens=True)
+            
+            scores = self.extract_scores(result)
+            sorted_scores = sorted(scores.items(), key=lambda item: item[1], reverse=True)
+            return sorted_scores
+
+        except Exception as e:
+            debug_info["stage"] = "processing model"
+            debug_info["error"] = str(e)
+            return debug_info
 
     def extract_scores(self, response):
         scores = {}