Update handler.py

handler.py

@@ -19,7 +19,7 @@ class EndpointHandler:
                 return_tensors="pt",
                 padding=True,
                 truncation=True,
-                max_length=512  # Réduit pour plus d'efficacité
+                max_length=512
             ).to(self.device)
             
             # Génération
@@ -35,36 +35,28 @@ class EndpointHandler:
             
             # Décodage
             translation = self.tokenizer.decode(generated_tokens[0], skip_special_tokens=True)
-            return translation
+            return [{"translation": translation}]
             
         except Exception as e:
-            return f"Error: {str(e)}"
+            return [{"error": str(e)}]
 
     def __call__(self, data):
         try:
             if "inputs" not in data:
-                raise ValueError("Request must contain 'inputs' field")
+                return [{"error": "Request must contain 'inputs' field"}]
             
             inputs = data["inputs"]
             
-            # Si l'entrée est une chaîne de caractères
+            # Que ce soit une chaîne ou une liste, on traite comme une seule entrée
             if isinstance(inputs, str):
-                translation = self.process_single_text(inputs)
-                return translation  # Retourne directement la traduction comme texte
-            
-            # Si l'entrée est une liste
-            elif isinstance(inputs, list):
-                translations = []
-                for item in inputs:
-                    if isinstance(item, dict) and "input" in item:
-                        translation = self.process_single_text(item["input"])
-                        translations.append({"output": translation})
-                    else:
-                        translations.append({"error": "Invalid input format"})
-                return translations
-            
+                return self.process_single_text(inputs)
+            elif isinstance(inputs, list) and len(inputs) > 0:
+                if isinstance(inputs[0], dict) and "input" in inputs[0]:
+                    return self.process_single_text(inputs[0]["input"])
+                else:
+                    return [{"error": "Invalid input format"}]
             else:
-                raise ValueError("Invalid input format")
+                return [{"error": "Invalid input format"}]
                 
         except Exception as e:
-            return str(e)  # Retourne l'erreur comme texte
+            return [{"error": str(e)}]