Update app.py

app.py

@@ -71,6 +71,27 @@ def get_image_ext(img_bytes):
 def diffuser_callback(i, t, latents):
     pass
 
+def preprocess_image(image):
+    w, h = image.size
+    w, h = map(lambda x: x - x % 32, (w, h))  # resize to integer multiple of 32
+    image = image.resize((w, h), resample=PIL.Image.LANCZOS)
+    image = np.array(image).astype(np.float32) / 255.0
+    image = image[None].transpose(0, 3, 1, 2)
+    image = torch.from_numpy(image)
+    return 2.0 * image - 1.0
+
+def preprocess_mask(mask):
+    mask = mask.convert("L")
+    w, h = mask.size
+    w, h = map(lambda x: x - x % 32, (w, h))  # resize to integer multiple of 32
+    mask = mask.resize((w // 8, h // 8), resample=PIL.Image.NEAREST)
+    mask = np.array(mask).astype(np.float32) / 255.0
+    mask = np.tile(mask, (4, 1, 1))
+    mask = mask[None].transpose(0, 1, 2, 3)  # what does this step do?
+    mask = 1 - mask  # repaint white, keep black
+    mask = torch.from_numpy(mask)
+    return mask
+    
 def process(init_image, mask):
     global model
     '''
@@ -188,7 +209,8 @@ def read_content(file_path: str) -> str:
 
     return content
 
-def predict(dict, prompt=""):
+def predict(dict):
+    print(f'liuyz_0_', dict)
     init_image = dict["image"] # .convert("RGB") #.resize((512, 512))
     print(f'liuyz_1_', init_image)
     print(f'liuyz_2_', init_image.convert("RGB"))
@@ -202,8 +224,8 @@ def predict(dict, prompt=""):
 
 css = '''
 .container {max-width: 1150px;margin: auto;padding-top: 1.5rem}
-#image_upload{min-height:400px}
-#image_upload [data-testid="image"], #image_upload [data-testid="image"] > div{min-height: 400px}
+#image_upload{min-height:auto}
+#image_upload [data-testid="image"], #image_upload [data-testid="image"] > div{min-height: auto}
 #mask_radio .gr-form{background:transparent; border: none}
 #word_mask{margin-top: .75em !important}
 #word_mask textarea:disabled{opacity: 0.3}
@@ -252,9 +274,9 @@ with image_blocks as demo:
                     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(
-                            margin=False,
-                            rounded=(False, True, True, False),
-                            full_width=False,
+                            margin=True,
+                            rounded=(True, True, True, True),
+                            full_width=True,
                         )
                 '''
                 with gr.Column():