Update app.py

app.py

@@ -220,15 +220,15 @@ def read_content(file_path: str) -> str:
     return content
 
 def predict(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))
-    '''
+    # print(f'liuyz_0_', dict)  
+     
+    image = dict["image"] # .convert("RGB") #.resize((512, 512))
+    # target_size = (init_image.shape[0], init_image.shape[1])
+    print(f'liuyz_1_', image.shpe)
+    print(f'liuyz_2_', image.convert("RGB").shape)
+    print(f'liuyz_3_', image.convert("RGB").resize((512, 512)).shape)
+    # mask = dict["mask"] # .convert("RGB") #.resize((512, 512))
+    
     image = Image.fromarray(dict["image"])
     mask = Image.fromarray(dict["mask"])
     output = model_process(dict["image"], dict["mask"])