chore: Update segformer_ckpt in app.py to use mattmdjaga/segformer_b2_clothes

app.py

@@ -125,7 +125,7 @@ pipeline = CatVTONPipeline(
 mask_processor = VaeImageProcessor(vae_scale_factor=8, do_normalize=False, do_binarize=True, do_convert_grayscale=True)
 automasker = AutoMaskerSeg(
     densepose_ckpt=os.path.join(repo_path, "DensePose"),
-    segformer_ckpt="/home/chongzheng_p23/data/Projects/CatVTON-main/Models/segformer_b3_clothes",
+    segformer_ckpt="mattmdjaga/segformer_b2_clothes",
     device='cuda', 
 )
 

requirements.txt

@@ -1,7 +1,5 @@
 accelerate==0.31.0
 diffusers==0.29.2
-gradio==4.39.0
-spaces==0.29.2
 huggingface_hub==0.23.4
 matplotlib==3.9.1
 numpy==1.26.4
@@ -11,10 +9,11 @@ PyYAML==6.0.1
 scipy==1.13.1
 setuptools==51.0.0
 scikit-image==0.24.0
-torch==2.0.1
-torchvision==0.15.2
+torch==2.1.2
+torchvision==0.16.2
 tqdm==4.66.4
 transformers==4.27.3
+xformers==0.0.23.post1
+Ninja==1.11.1.1
 git+https://github.com/facebookresearch/detectron2.git
-git+https://github.com/facebookresearch/detectron2@main#subdirectory=projects/DensePose
-Ninja==1.11.1
+git+https://github.com/facebookresearch/detectron2@main#subdirectory=projects/DensePose