Update app.py

app.py

@@ -124,7 +124,14 @@ pipe.unet_encoder = UNet_Encoder
 @spaces.GPU
 def start_tryon(dict,garm_img,garment_des,is_checked,is_checked_crop,denoise_steps,seed, category):
     device = "cuda"
+    category = int(category) 
+    if category==0:
+        category='upper_body'
     
+    elif category==1:
+        category='lower_body'
+    else:
+        category='dresses'
     openpose_model.preprocessor.body_estimation.model.to(device)
     pipe.to(device)
     pipe.unet_encoder.to(device)