fix bugradio (#455)

2023-09-20 06:00:52 -07:00 · 2023-09-20 06:00:52 -07:00 · 6597b3df64
commit 6597b3df64
parent 5456a5784b
1 changed files with 4 additions and 3 deletions
--- a/modules/gradio_hijack.py
+++ b/modules/gradio_hijack.py
@ -265,10 +265,11 @@ class Image(
        if x is None:
            return x

-        mask = ""
+        mask = None
+
        if self.tool == "sketch" and self.source in ["upload", "webcam"]:
-            assert isinstance(x, dict)
-            x, mask = x["image"], x["mask"]
+            if isinstance(x, dict):
+                x, mask = x["image"], x["mask"]

        assert isinstance(x, str)
        im = processing_utils.decode_base64_to_image(x)