Upload app.py

app.py

@@ -49,19 +49,19 @@ parser.add_argument('--if_ipa', type=bool, default=True)
 parser.add_argument('--if_control', type=bool, default=True)
 
 parser.add_argument('--pretrained_model_name_or_path',
-                    default="/home/sf/Realistic_Vision_V4.0_noVAE",
+                    default="./ckpt/Realistic_Vision_V4.0_noVAE",
                     type=str)
 parser.add_argument('--ip_ckpt',
-                    default="/home/sf/ip_adapter/ip-adapter-faceid-plus_sd15.bin",
+                    default="./ckpt/ip-adapter-faceid-plus_sd15.bin",
                     type=str)
 parser.add_argument('--pretrained_image_encoder_path',
-                    default="/home/sf/ip_adapter/image_encoder/",
+                    default="./ckpt/image_encoder/",
                     type=str)
 parser.add_argument('--pretrained_vae_model_path',
-                    default="/home/sf/sd-vae-ft-mse/",
+                    default="./ckpt/sd-vae-ft-mse/",
                     type=str)
 parser.add_argument('--model_ckpt',
-                    default="/home/sf/weights/sd_stage2/mp_rank_00_model_states_0628.pt",
+                    default="/./ckpt/IMAGDressing-v1_512.pt",
                     type=str)
 parser.add_argument('--output_path', type=str, default="./output_ipa_control_resampler")
 # parser.add_argument('--device', type=str, default="cuda:0")