Update handler.py

handler.py

@@ -55,17 +55,15 @@ class EndpointHandler():
 
     def __call__(self, data: Dict[str, Any]) -> str:
         # get inputs 
-        data = data.pop("inputs", data)
-        data = data.decode('utf-8')
-        data = json.loads(data)
+        contexts = data.pop("inputs", data)
         
         batch_size = data.pop("batch_size", 1)
 
-        context = base64.b64decode(data["context_b64str"])
+        context = base64.b64decode(contexts[0])
         context = np.frombuffer(context, dtype="float32")
         context = np.reshape(context, (batch_size, 77, 768))
 
-        unconditional_context = base64.b64decode(data['unconditional_context_b64str'])
+        unconditional_context = base64.b64decode(contexts[1])
         unconditional_context = np.frombuffer(unconditional_context, dtype="float32")
         unconditional_context = np.reshape(unconditional_context, (batch_size, 77, 768))        
 
@@ -98,4 +96,4 @@ class EndpointHandler():
         latent_b64 = base64.b64encode(latent.numpy().tobytes())
         latent_b64str = latent_b64.decode()
 
-        return latent_b64str
+        return latent_b64str