Upload app.py

app.py

@@ -121,9 +121,9 @@ pipe = TryonPipeline.from_pretrained(
 )
 pipe.unet_encoder = UNet_Encoder
 
-@spaces.GPU
+@spaces.CPU
 def start_tryon(dict,garm_img,garment_des,is_checked,is_checked_crop,denoise_steps,seed):
-    device = "cuda"
+    device = "cpu"
     
     openpose_model.preprocessor.body_estimation.model.to(device)
     pipe.to(device)
@@ -211,20 +211,20 @@ def start_tryon(dict,garm_img,garment_des,is_checked,is_checked_crop,denoise_ste
 
 
 
-                    pose_img =  tensor_transfrom(pose_img).unsqueeze(0).to(device,torch.float16)
-                    garm_tensor =  tensor_transfrom(garm_img).unsqueeze(0).to(device,torch.float16)
+                    pose_img =  tensor_transfrom(pose_img).unsqueeze(0).to(device)
+                    garm_tensor =  tensor_transfrom(garm_img).unsqueeze(0).to(device)
                     generator = torch.Generator(device).manual_seed(seed) if seed is not None else None
                     images = pipe(
-                        prompt_embeds=prompt_embeds.to(device,torch.float16),
-                        negative_prompt_embeds=negative_prompt_embeds.to(device,torch.float16),
-                        pooled_prompt_embeds=pooled_prompt_embeds.to(device,torch.float16),
-                        negative_pooled_prompt_embeds=negative_pooled_prompt_embeds.to(device,torch.float16),
+                        prompt_embeds=prompt_embeds.to(device),
+                        negative_prompt_embeds=negative_prompt_embeds.to(device),
+                        pooled_prompt_embeds=pooled_prompt_embeds.to(device),
+                        negative_pooled_prompt_embeds=negative_pooled_prompt_embeds.to(device),
                         num_inference_steps=denoise_steps,
                         generator=generator,
                         strength = 1.0,
-                        pose_img = pose_img.to(device,torch.float16),
-                        text_embeds_cloth=prompt_embeds_c.to(device,torch.float16),
-                        cloth = garm_tensor.to(device,torch.float16),
+                        pose_img = pose_img.to(device),
+                        text_embeds_cloth=prompt_embeds_c.to(device),
+                        cloth = garm_tensor.to(device),
                         mask_image=mask,
                         image=human_img, 
                         height=1024,