deploy

app.py

@@ -129,6 +129,7 @@ POSE_HEIGHT = int(HEIGHT/4)  #int(HEIGHT/2)
 @spaces.GPU
 def start_tryon(dict,garm_img,garment_des,is_checked,is_checked_crop,denoise_steps,seed):
     device = "cuda"
+    # device = 'cuda:0' if torch.cuda.is_available() else 'cpu'
     
     openpose_model.preprocessor.body_estimation.model.to(device)
     pipe.to(device)