Update handler.py

handler.py

@@ -23,6 +23,5 @@ class EndpointHandler():
             raw_image = Image.open(requests.get(url, stream=True).raw)
             inputs = self.processor(prompt, raw_image, return_tensors='pt').to(0, torch.float16)
             output = self.model.generate(**inputs, max_new_tokens=200, do_sample=False)
-            # Convert Tensor to NumPy array or list before returning
-            output = output.cpu().numpy().tolist()
+            print(self.processor.decode(output[0][2:], skip_special_tokens=True))
         return output