Update handler.py

handler.py

@@ -1,6 +1,7 @@
 import os
 import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
+from typing import Dict, List, Any # <-- ADD THIS LINE
 
 class EndpointHandler():
     def __init__(self, model_id: str):
@@ -48,6 +49,7 @@ class EndpointHandler():
         print(f"Configured label order for output: {self.label_names}")
 
 
+    # Corrected type hints in the signature below
     def __call__(self, data: Dict[str, Any]) -> Dict[str, Any] | List[Dict[str, Any]]:
         """
         Handles inference requests.