add custom handler

handler.py

@@ -22,9 +22,11 @@ class EndpointHandler():
     def __call__(self, data: Dict[str, Any]) -> str:
         # get inputs 
         latent = data.pop("inputs", data)
+        batch_size = data.pop("batch_size", 1)
+
         latent = base64.b64decode(latent)
         latent = np.frombuffer(latent, dtype="float32")
-        latent = np.reshape(latent, (1, 64, 64, 4))
+        latent = np.reshape(latent, (batch_size, 64, 64, 4))
 
         decoded = self.decoder.predict_on_batch(latent)
         decoded = ((decoded + 1) / 2) * 255

text-encoder

@@ -0,0 +1 @@
+Subproject commit e06cb5f39e98f1035306142bb72769ce1ba93d73