Keshabwi66 commited on
Commit
65e7974
·
verified ·
1 Parent(s): ec0b3ac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -93,8 +93,7 @@ UNet_Encoder = UNet2DConditionModel_ref.from_pretrained(
93
  torch_dtype=torch.float16,
94
  )
95
 
96
- parsing_model = Parsing(0)
97
- openpose_model = OpenPose(0)
98
 
99
  UNet_Encoder.requires_grad_(False)
100
  image_encoder.requires_grad_(False)
@@ -127,7 +126,6 @@ pipe.unet_encoder = UNet_Encoder
127
  @spaces.GPU
128
  def start_tryon(person_img, pose_img, mask_img, cloth_img, garment_des, denoise_steps, seed):
129
  # Assuming device is set up (e.g., "cuda" or "cpu")
130
- openpose_model.preprocessor.body_estimation.model.to(device)
131
  pipe.to(device)
132
  pipe.unet_encoder.to(device)
133
 
 
93
  torch_dtype=torch.float16,
94
  )
95
 
96
+
 
97
 
98
  UNet_Encoder.requires_grad_(False)
99
  image_encoder.requires_grad_(False)
 
126
  @spaces.GPU
127
  def start_tryon(person_img, pose_img, mask_img, cloth_img, garment_des, denoise_steps, seed):
128
  # Assuming device is set up (e.g., "cuda" or "cpu")
 
129
  pipe.to(device)
130
  pipe.unet_encoder.to(device)
131