Update app.py

app.py

@@ -239,6 +239,10 @@ def inpaint(image, invert):
     result = np.concatenate(color)
     return [result]
 
+
+def predict(im):
+    return im["composite"]
+
 block = gr.Blocks().queue()
 with block:
     gr.Markdown(DESCRIPTION)
@@ -265,6 +269,8 @@ with block:
     with gr.Row():
         with gr.Column():
             input_image = gr.ImageMask(sources="upload", type="numpy", height="auto")
+            im_preview = gr.Image()
+            input_image.change(predict, outputs=im_preview, inputs=input_image, show_progress="hidden")
 #            input_image = gr.ImageEditor(sources="upload", type="numpy", height="auto", layers="False", brush=gr.Brush(colors=["#000000"]))
 #            input_image = gr.Image(source='upload', type="numpy", tool="sketch", height=512)
 #            resolution = gr.Slider(label="resolution", minimum=256, maximum=1024, value=512, step=64)