Zhibinhong
/

Instruct-Pix2pix

Inference Endpoints

Model card Files Files and versions Community

Zhibinhong commited on Apr 25, 2023

Commit

9c246cc

•

1 Parent(s): c968655

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -15,7 +15,7 @@ class EndpointHandler():
         info=data['inputs']
         image=info.pop("image",data)
-        prompt=info.pop("prompt",data).decode('utf-8')
         image=base64.b64decode(image)
         raw_images = Image.open(BytesIO(image)).convert('RGB')

         info=data['inputs']
         image=info.pop("image",data)
+        prompt=info.pop("prompt",data)
         image=base64.b64decode(image)
         raw_images = Image.open(BytesIO(image)).convert('RGB')