Update handler.py

handler.py

@@ -61,4 +61,4 @@ class EndpointHandler():
         #     )
         # captions = self.processor.batch_decode(out, skip_special_tokens=True)
 
-        return {"captions": captions}
+        return {"captions": generated_caption}