Spaces:

myn0908
/

S2I-Artwork-Sketch-to-Image-Diffusion

Running on Zero

myn0908 commited on 29 days ago

Commit

eb08382

•

1 Parent(s): adbcf49

update code

Files changed (1) hide show

app.py CHANGED Viewed

@@ -48,7 +48,8 @@ def clear_session(image):
     global clear_flag
     clear_flag = True
     if image is not None:
-        return gr.update(value=None), gr.update(value=None)
     return gr.update(value=None), gr.update(value=None)

     global clear_flag
     clear_flag = True
     if image is not None:
+        ones = Image.new("L", (512, 512), 255)
+        return ones, gr.update(value=None)
     return gr.update(value=None), gr.update(value=None)