Support for zeroGPU-a

README.md

@@ -1,15 +1,14 @@
 ---
-title: GEN10 IDM-VTON Base
-emoji: ⚡
-colorFrom: gray
-colorTo: yellow
+title: IDM VTON BASE
+emoji: 👕👔👚
+colorFrom: yellow
+colorTo: red
 sdk: gradio
-sdk_version: 4.44.0
-python_version: 3.10.13
+sdk_version: 4.24.0
 app_file: app.py
 pinned: false
-license: apache-2.0
-short_description: 'Gen10조 Base'
+license: cc-by-nc-sa-4.0
+short_description: IDM VTOM BASE
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -123,6 +123,8 @@ pipe = TryonPipeline.from_pretrained(
 )
 pipe.unet_encoder = UNet_Encoder
 
+
+@spaces.GPU
 def start_tryon(dict,garm_img,garment_des,is_checked,is_checked_crop,denoise_steps,seed):
     
     openpose_model.preprocessor.body_estimation.model.to(device)

preprocess/openpose/run_openpose.py

@@ -28,12 +28,12 @@ import pdb
 
 class OpenPose:
     def __init__(self, gpu_id: int):
-        self.gpu_id = gpu_id
-        torch.cuda.set_device(gpu_id)
+        # self.gpu_id = gpu_id
+        # torch.cuda.set_device(gpu_id)
         self.preprocessor = OpenposeDetector()
 
     def __call__(self, input_image, resolution=384):
-        torch.cuda.set_device(self.gpu_id)
+        # torch.cuda.set_device(self.gpu_id)
         if isinstance(input_image, Image.Image):
             input_image = np.asarray(input_image)
         elif type(input_image) == str:

requirements.txt

@@ -1,10 +1,8 @@
-huggingface_hub~=0.23.0
-gradio==4.44.0
-transformers==4.36.2
 torch==2.1.0
 torchvision==0.16.0
 torchaudio==2.1.0
 numpy==1.24.4
+pydantic==2.10.6
 scipy==1.10.1
 scikit-image==0.21.0
 opencv-python==4.7.0.72
@@ -17,11 +15,10 @@ tqdm==4.64.1
 config==0.5.1
 einops==0.7.0
 onnxruntime==1.16.2
-datasets==2.19.0
-triton==2.1.0
 basicsr
 av
 fvcore
 cloudpickle
 omegaconf
-pycocotools
+pycocotools
+huggingface_hub==0.25.0