Upload app.py

app.py

@@ -74,9 +74,6 @@ args = parser.parse_args()
 
 args.device = "cuda"
 
-base_path = 'feishen29/IMAGDressing-v1'
-
-
 vae = AutoencoderKL.from_pretrained('stabilityai/sd-vae-ft-mse').to(dtype=torch.float16, device=args.device)
 tokenizer = CLIPTokenizer.from_pretrained("SG161222/Realistic_Vision_V4.0_noVAE", subfolder="tokenizer")
 text_encoder = CLIPTextModel.from_pretrained("SG161222/Realistic_Vision_V4.0_noVAE", subfolder="text_encoder").to(dtype=torch.float16, device=args.device)
@@ -292,24 +289,19 @@ def dress_process(garm_img, face_img, pose_img, prompt, cloth_guidance_scale, ca
  # return result[OutputKeys.OUTPUT_IMG]
  return output[0]
 
-example_path = os.path.dirname(__file__)
-
-garm_list = os.listdir(os.path.join(example_path, "cloth", 'cloth'))
-garm_list_path = [os.path.join(example_path, "cloth", 'cloth', garm) for garm in garm_list]
-
-face_list = os.listdir(os.path.join(example_path, "face", 'face'))
-face_list_path = [os.path.join(example_path, "face", 'face', face) for face in face_list]
-
-pose_list = os.listdir(os.path.join(example_path, "pose", 'pose'))
-pose_list_path = [os.path.join(example_path, "pose", 'pose', pose) for pose in pose_list]
+base_path = 'yisol/IDM-VTON'
+example_path = os.path.join(os.path.dirname(__file__), 'example')
 
+garm_list = os.listdir(os.path.join(example_path,"cloth"))
+garm_list_path = [os.path.join(example_path,"cloth",garm) for garm in garm_list]
 
+face_list = os.listdir(os.path.join(example_path,"face"))
+face_list_path = [os.path.join(example_path,"face",face) for face in face_list]
 
-##default human
+pose_list = os.listdir(os.path.join(example_path,"pose"))
+pose_list_path = [os.path.join(example_path,"pose",pose) for pose in pose_list]
 
-def process_image(image):
 
- return image
 
 image_blocks = gr.Blocks().queue()
 with image_blocks as demo:
@@ -335,8 +327,6 @@ with image_blocks as demo:
  example = gr.Examples(
  inputs=imgs,
  examples_per_page=10,
- fn=process_image,
- outputs=imgs,
  examples=face_list_path
  )
  with gr.Row():
@@ -350,8 +340,6 @@ with image_blocks as demo:
  example = gr.Examples(
  inputs=pose_img,
  examples_per_page=8,
- fn=process_image,
- outputs=pose_img,
  examples=pose_list_path)
 
  # with gr.Column():