Update app.py

app.py

@@ -212,15 +212,19 @@ def read_content(file_path: str) -> str:
     return content
 
 def predict(dict):
-    print(f'liuyz_0_', dict)
+    '''
+    print(f'liuyz_0_', dict)    
     init_image = dict["image"] # .convert("RGB") #.resize((512, 512))
     target_size = (init_image.shape[0], init_image.shape[1])
     print(f'liuyz_1_', init_image, target_size)
     print(f'liuyz_2_', init_image.convert("RGB").shape)
     print(f'liuyz_3_', init_image.convert("RGB").resize((512, 512)).shape)
     mask = dict["mask"] # .convert("RGB") #.resize((512, 512))
-    # output = model_process(init_image, mask)
-    output = None # output.images[0]
+    '''
+    image = Image.fromarray(dict["image"])
+    mask= Image.fromarray(dict["mask"])
+    output = model_process(image, mask)
+    output = output.images[0]
     # output = pipe(prompt = prompt, image=init_image, mask_image=mask,guidance_scale=7.5)
     
     return output #, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)