Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -93,8 +93,7 @@ UNet_Encoder = UNet2DConditionModel_ref.from_pretrained(
|
|
93 |
torch_dtype=torch.float16,
|
94 |
)
|
95 |
|
96 |
-
|
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 |
|