Update run/gradio_ootd.py

run/gradio_ootd.py

@@ -62,16 +62,17 @@ def decode_image_from_base64(image_data):
 def process_hd(vton_img, garm_img,category, n_samples, n_steps, image_scale, seed):
     model_type = 'dc'
     with torch.no_grad():
-        openpose_model_dc.preprocessor.body_estimation.model.toopenpose_model_dc('cuda')
-        ootd_model_hd.pipe.to('cuda')
-        ootd_model_hd.image_encoder.to('cuda')
-        ootd_model_hd.text_encoder.to('cuda')
-        
+        openpose_model_dc.preprocessor.body_estimation.model.to('cuda')
+        ootd_model_dc.pipe.to('cuda')
+        ootd_model_dc.image_encoder.to('cuda')
+        ootd_model_dc.text_encoder.to('cuda')
+
         garm_img = decode_image_from_base64(garm_img).resize((768, 1024))
         vton_img = decode_image_from_base64(vton_img).resize((768, 1024))
         keypoints = openpose_model_dc(vton_img.resize((384, 512)))
         model_parse, _ = parsing_model_dc(vton_img.resize((384, 512)))
 
+
         mask, mask_gray = get_mask_location(model_type, category_dict_utils[category], model_parse, keypoints)
         mask = mask.resize((768, 1024), Image.NEAREST)
         mask_gray = mask_gray.resize((768, 1024), Image.NEAREST)