keras-sd
/

text-encoder

Model card Files Files and versions

chansung commited on Dec 17, 2022

Commit

d063207

·

1 Parent(s): a57f986

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -59,7 +59,7 @@ class EndpointHandler():
         return context, unconditional_context
-    def __call__(self, data: Dict[str, Any]) -> str:
         # get inputs
         prompt = data.pop("inputs", data)
         batch_size = data.pop("batch_size", 1)

         return context, unconditional_context
+    def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
         # get inputs
         prompt = data.pop("inputs", data)
         batch_size = data.pop("batch_size", 1)