fixed arg error

handler.py

@@ -165,7 +165,7 @@ class EndpointHandler:
         # create inference pipeline
 
 
-    def __call__(self, data: Any) -> List[List[Dict[str, float]]]:
+    def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         """
         Args:
             data (:obj:):
@@ -175,7 +175,9 @@ class EndpointHandler:
                 - "label": A string representing what the label/class is. There can be multiple labels.
                 - "score": A score between 0 and 1 describing how confident the model is for this label/class.
         """
-
+        print("data", data)
+        inputs = data.pop("inputs", data)
+        start_string = inputs[0]
         config = {
             "vocab_size": vocab_size,
             "embedding_dim": embedding_dim,
@@ -201,7 +203,7 @@ class EndpointHandler:
         for i in range(10):
             error_rate, generated_text = generate_text_bpe(
                 model,
-                start_string=data,
+                start_string=start_string,
                 generation_length=seq_length,
                 temperature=1.2,
                 top_k=20,