Update app.py

app.py

@@ -71,8 +71,8 @@ def process(query_img, state,
     cfg = Config.fromfile(cfg_path)
     width, height, _ = state['original_support_image'].shape
     kp_src_np = np.array(state['kp_src']).copy().astype(np.float32)
-    kp_src_np[:, 0] = kp_src_np[:,0] / (width // 2) * cfg.model.encoder_config.img_size
-    kp_src_np[:, 1] = kp_src_np[:,1] / (height // 2) * cfg.model.encoder_config.img_size
+    kp_src_np[:, 0] = kp_src_np[:,0] / (width // 4) * cfg.model.encoder_config.img_size
+    kp_src_np[:, 1] = kp_src_np[:,1] / (height // 4) * cfg.model.encoder_config.img_size
     kp_src_np = np.flip(kp_src_np, 1).copy()
     kp_src_tensor = torch.tensor(kp_src_np).float()
     preprocess = transforms.Compose([
@@ -296,7 +296,7 @@ with gr.Blocks() as demo:
         state['kp_src'].clear()
         state['original_support_image'] = np.array(support_img)[:, :, ::-1].copy()
         width, height = support_img.size
-        support_img = support_img.resize((width // 2, width // 2), Image.Resampling.LANCZOS)
+        support_img = support_img.resize((width // 4, width // 4), Image.Resampling.LANCZOS)
         return support_img, support_img, state