Update handler.py

handler.py

@@ -12,11 +12,11 @@ class EndpointHandler():
         )
         self.pipeline = pipeline("image-to-text", model=path, model_kwargs={"quantization_config": quantization_config})
 
-    def __call__(self, data: Dict[str, Any]):
+    def __call__(self, data: Dict[Any, Any]):
         totalarr = []
         inputs = data.pop("inputs", '')
         prompt_base = data.pop("prompt", "")
         image = Image.open(requests.get(inputs, stream=True).raw)
         prompt = f"USER: <image>\n{prompt_base}.Answer in one word\nASSISTANT:"
         outputs = self.pipeline(image, prompt=prompt, generate_kwargs={"max_new_tokens": 200})
-        return (outputs[0]['generated_text'])  # Extract the generated_text from the pipeline output
+        return (outputs[0]['generated_text'])  # Extract the generated_text from the pipeline output