Update handler.py

handler.py

@@ -7,13 +7,26 @@ class EndpointHandler:
 
     def load(self):
         print("Model loading skipped for Hello World API.")
-        return
 
-    def predict(self, inputs):
-        if not isinstance(inputs, dict) or "inputs" not in inputs:
-            return {"error": "Invalid input format. Expected {'inputs': 'your text'}"}
+    def __call__(self, inputs):
+        """
+        Hugging Face expects the `EndpointHandler` to be callable.
+        So, we define `__call__` instead of `predict`.
+        """
+        try:
+            # Hugging Face sends input as a LIST or DICT, so handle both cases
+            if isinstance(inputs, list) and len(inputs) > 0:
+                user_text = inputs[0]  # Extract text from the list
+            elif isinstance(inputs, dict) and "inputs" in inputs:
+                user_text = inputs["inputs"]
+            else:
+                return {"error": "Invalid input format. Expected {'inputs': 'your text'} or ['your text']."}
 
-        user_text = inputs["inputs"]
-        current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+            # Generate timestamp
+            current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
 
-        return {"message": f"Received at {current_time}: {user_text}"}
+            # Return formatted message
+            return {"message": f"Received at {current_time}: {user_text}"}
+
+        except Exception as e:
+            return {"error": f"Unexpected error: {str(e)}"}