Update app.py

app.py

@@ -355,11 +355,11 @@ def predict(input):
     print(f'liuyz_3_', image.convert("RGB").resize((512, 512)).shape)
     # mask = dict["mask"] # .convert("RGB") #.resize((512, 512))
     '''   
-    output = model_process(input) # dict["image"], dict["mask"])
+    # output = model_process(input) # dict["image"], dict["mask"])
     
     # output = mask #output.images[0]
     # output = pipe(prompt = prompt, image=init_image, mask_image=mask,guidance_scale=7.5)
-    # output = input["mask"]
+    output = input["mask"]
     return output #, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
 
 print(f'liuyz_500_here_')
@@ -452,7 +452,7 @@ with image_blocks as demo:
         with gr.Box():
             with gr.Row():
                 with gr.Column():
-                    image = gr.Image(source='upload', tool='sketch', type='filepath',label="Upload").style(height=512)
+                    image = gr.Image(source='upload', tool='sketch',label="Upload").style(height=512)
                     with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True):
                         # prompt = gr.Textbox(placeholder = 'Your prompt (what you want in place of what is erased)', show_label=False, elem_id="input-text")
                         btn = gr.Button("Done!").style(