Update app.py

app.py

@@ -105,7 +105,7 @@ def preprocess_mask(mask):
     return mask
     
 model = None
-def model_process(init_image, mask):
+def model_process(image, mask):
     global model
     
     # input = request.files
@@ -117,11 +117,12 @@ def model_process(init_image, mask):
     
     # image, alpha_channel = load_img(origin_image_bytes)
     # Origin image shape: (512, 512, 3)
-    original_shape = init_image.shape
+    original_shape = image.shape
     interpolation = cv2.INTER_CUBIC    
     
     # form = request.form
-    
+    print(f'liuyz_3_here_', original_shape)
+
     size_limit = 1080 # : Union[int, str] = form.get("sizeLimit", "1080")
     if size_limit == "Original":
         size_limit = max(image.shape)
@@ -151,22 +152,24 @@ def model_process(init_image, mask):
         cv2_flag='INPAINT_NS',
         cv2_radius=5,
     )
+    print(f'config = {config}')
 
-    if config.sd_seed == -1:
-        config.sd_seed = random.randint(1, 999999999)
-
+    # if config.sd_seed == -1:
+    #    config.sd_seed = random.randint(1, 999999999)
+    
+    config.sd_seed = 42
     # logger.info(f"Origin image shape: {original_shape}")
     print(f"Origin image shape: {original_shape}")
     image = resize_max_size(image, size_limit=size_limit, interpolation=interpolation)
     # logger.info(f"Resized image shape: {image.shape}")
     print(f"Resized image shape: {image.shape}")
     
+    #mask, _ = load_img(input["mask"].read(), gray=True)
+    mask = resize_max_size(mask, size_limit=size_limit, interpolation=interpolation)
+
     if model is None:
         return None
         
-    mask, _ = load_img(input["mask"].read(), gray=True)
-    mask = resize_max_size(mask, size_limit=size_limit, interpolation=interpolation)
-
     start = time.time()
     res_np_img = model(image, mask, config)
     logger.info(f"process time: {(time.time() - start) * 1000}ms")