simulationcartridge
/

ppl

stable-diffusion

template:sd-lora

Model card Files Files and versions

Mirza Učanbarlić commited on Nov 11, 2023

Commit

f0884e6

·

1 Parent(s): dcdeaa3

update handler

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -22,7 +22,7 @@ class EndpointHandler():
          Return:
             A :obj:`list` | `dict`: will be serialized and returned
         """
-        inputs = data.pop("input_image", data)
         prompt = data.pop("prompt", None)
         # run normal prediction

          Return:
             A :obj:`list` | `dict`: will be serialized and returned
         """
+        input_image= data.pop("input_image", data)
         prompt = data.pop("prompt", None)
         # run normal prediction