cloth_try_on

README.md

@@ -4,7 +4,7 @@ emoji: 👕👔👚
 colorFrom: yellow
 colorTo: red
 sdk: gradio
-sdk_version: 5.25.2
+sdk_version: 4.24.0
 app_file: app.py
 pinned: false
 license: cc-by-nc-sa-4.0

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-# import spaces
+import spaces
 from PIL import Image
 from src.tryon_pipeline import StableDiffusionXLInpaintPipeline as TryonPipeline
 from src.unet_hacked_garmnet import UNet2DConditionModel as UNet2DConditionModel_ref
@@ -121,7 +121,7 @@ pipe = TryonPipeline.from_pretrained(
 )
 pipe.unet_encoder = UNet_Encoder
 
-# @spaces.GPU
+@spaces.GPU
 def start_tryon(dict,garm_img,garment_des,is_checked,is_checked_crop,denoise_steps,seed):
     device = "cuda"
     

requirements.txt

@@ -1,8 +1,8 @@
-torch==2.1.0
-torchvision==0.16.0
-torchaudio==2.1.0
+transformers==4.36.2
+torch==2.0.1
+torchvision==0.15.2
+torchaudio==2.0.2
 numpy==1.24.4
-pydantic==2.10.6
 scipy==1.10.1
 scikit-image==0.21.0
 opencv-python==4.7.0.72
@@ -20,5 +20,4 @@ av
 fvcore
 cloudpickle
 omegaconf
-pycocotools
-huggingface_hub==0.25.0
+pycocotools