path changed

README.md

@@ -9,5 +9,3 @@ pinned: false
 ---
 
 # FastAPI-Deployment
-
-Simple Text Generation app using Cohere generate a model to demonstrate deploying FastAPI Applications on Huggingface Spaces via Docker and Github Actions

app.py

@@ -52,7 +52,7 @@ def pil_to_binary_mask(pil_image, threshold=0):
     return output_mask
 
 
-base_path = 'nami0342/GenAI_VTON_API'
+base_path = 'yisol/IDM-VTON'  # Reverting to original model path
 example_path = os.path.join(os.path.dirname(__file__), 'example')
 
 unet = UNet2DConditionModel.from_pretrained(
@@ -182,10 +182,8 @@ def start_tryon(humanTarget_img,garm_img,garment_des,is_automaskchecked,is_check
         # #     mask, mask_gray = get_mask_location('hd', "dresses", model_parse, keypoints)
         # #     mask = mask.resize((768,1024))
         # else:
-        mask_temp_img = Image()
+        mask_temp_img = Image.new('RGB', (768, 1024), color='white')  # Creating a new blank image
         mask = pil_to_binary_mask(mask_temp_img.convert("RGB").resize((768, 1024)))
-            # mask = transforms.ToTensor()(mask)
-            # mask = mask.unsqueeze(0)
     mask_gray = (1-transforms.ToTensor()(mask)) * tensor_transfrom(human_img)
     mask_gray = to_pil_image((mask_gray+1.0)/2.0)