Skip to content

Commit

Permalink
chore: Add ZeroGPU HuggingFace Space link
Browse files Browse the repository at this point in the history
  • Loading branch information
Zheng-Chong committed Aug 10, 2024
1 parent 5bad478 commit fe6be79
Show file tree
Hide file tree
Showing 9 changed files with 24 additions and 5 deletions.
13 changes: 11 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,9 @@
<a href="http://120.76.142.206:8888" style="margin: 0 2px;">
<img src='https://img.shields.io/badge/Demo-Gradio-gold?style=flat&logo=Gradio&logoColor=red' alt='Demo'>
</a>
<a href="https://huggingface.co/spaces/zhengchong/CatVTON" style="margin: 0 2px;">
<img src='https://img.shields.io/badge/Space-ZeroGPU-orange?style=flat&logo=Gradio&logoColor=red' alt='Demo'>
</a>
<a href='https://zheng-chong.github.io/CatVTON/' style="margin: 0 2px;">
<img src='https://img.shields.io/badge/Webpage-Project-silver?style=flat&logo=&logoColor=orange' alt='webpage'>
</a>
Expand All @@ -21,22 +24,28 @@
</a>
</div>



**CatVTON** is a simple and efficient virtual try-on diffusion model with ***1) Lightweight Network (899.06M parameters totally)***, ***2) Parameter-Efficient Training (49.57M parameters trainable)*** and ***3) Simplified Inference (< 8G VRAM for 1024X768 resolution)***.
<div align="center">
<img src="resource/img/teaser.jpg" width="100%" height="100%"/>
<img src="resource/img/teaser.jpg" width="85%" height="100%"/>
</div>


**CatVTON** is a simple and efficient virtual try-on diffusion model with ***1) Lightweight Network (899.06M parameters totally)***, ***2) Parameter-Efficient Training (49.57M parameters trainable)*** and ***3) Simplified Inference (< 8G VRAM for 1024X768 resolution)***.


## Updates
- **`2024/08/10`**: Our 🤗 [**HuggingFace Space**](https://huggingface.co/spaces/zhengchong/CatVTON) is available now! Thanks for the grant from [**ZeroGPU**](https://huggingface.co/zero-gpu-explorers)
- **`2024/08/09`**: [**Evaluation code**](https://github.com/Zheng-Chong/CatVTON?tab=readme-ov-file#3-calculate-metrics) is provided to calculate metrics 📚.
- **`2024/07/27`**: We provide code and workflow for deploying CatVTON on [**ComfyUI**](https://github.com/Zheng-Chong/CatVTON?tab=readme-ov-file#comfyui-workflow) 💥.
- **`2024/07/24`**: Our [**Paper on ArXiv**](http://arxiv.org/abs/2407.15886) is available 🥳!
- **`2024/07/22`**: Our [**App Code**](https://github.com/Zheng-Chong/CatVTON/blob/main/app.py) is released, deploy and enjoy CatVTON on your mechine 🎉!
- **`2024/07/21`**: Our [**Inference Code**](https://github.com/Zheng-Chong/CatVTON/blob/main/inference.py) and [**Weights** 🤗](https://huggingface.co/zhengchong/CatVTON) are released.
- **`2024/07/11`**: Our [**Online Demo**](http://120.76.142.206:8888) is released 😁.




## Installation
An [Installation Guide](https://github.com/Zheng-Chong/CatVTON/blob/main/INSTALL.md) is provided to help build the conda environment for CatVTON. When deploying the app, you will need Detectron2 & DensePose, which are not required for inference on datasets. Install the packages according to your needs.

Expand Down
Binary file modified __pycache__/utils.cpython-39.pyc
Binary file not shown.
6 changes: 3 additions & 3 deletions app.py
Original file line number Diff line number Diff line change
Expand Up @@ -119,7 +119,6 @@ def image_grid(imgs, rows, cols):
device='cuda',
)

@spaces.GPU
def submit_function(
person_image,
cloth_image,
Expand Down Expand Up @@ -228,9 +227,10 @@ def person_example_fn(image_path):
</a>
</div>
<br>
· This demo and our weights are only open for **Non-commercial Use**. <br>
· This demo and our weights are only for <span>Non-commercial Use</span>. <br>
· You can try CatVTON in our <a href="https://huggingface.co/spaces/zhengchong/CatVTON">HuggingFace Space</a> or our <a href="http://120.76.142.206:8888">online demo</a> (run on 3090). <br>
· Thanks to <a href="https://huggingface.co/zero-gpu-explorers">ZeroGPU</a> for providing A100 for our <a href="https://huggingface.co/spaces/zhengchong/CatVTON">HuggingFace Space</a>. <br>
· SafetyChecker is set to filter NSFW content, but it may block normal results too. Please adjust the <span>`seed`</span> for normal outcomes.<br>
· Thanks to <a href="https://huggingface.co/zero-gpu-explorers">ZeroGPU</a> for providing GPU for <a href="https://huggingface.co/spaces/zhengchong/CatVTON">Our HuggingFace Space.</a>
"""

def app_gradio():
Expand Down
10 changes: 10 additions & 0 deletions index.html
Original file line number Diff line number Diff line change
Expand Up @@ -108,6 +108,16 @@ <h1 class="title is-1 publication-title">🐈 CatVTON: Concatenation Is All You
<span>Demo</span>
</a>
</span>
<!-- Demo Link. -->
<span class="link-block">
<a href="https://huggingface.co/spaces/zhengchong/CatVTON"
class="external-link button is-normal is-rounded is-dark">
<span class="icon">
<i class="fas fa-gamepad"></i>
</span>
<span>Space</span>
</a>
</span>
<!-- Models Link. -->
<span class="link-block">
<a href="https://huggingface.co/zhengchong/CatVTON"
Expand Down
Binary file modified model/SCHP/modules/__pycache__/__init__.cpython-39.pyc
Binary file not shown.
Binary file modified model/SCHP/modules/__pycache__/bn.cpython-39.pyc
Binary file not shown.
Binary file modified model/SCHP/modules/__pycache__/functions.cpython-39.pyc
Binary file not shown.
Binary file modified model/SCHP/networks/__pycache__/__init__.cpython-39.pyc
Binary file not shown.
Binary file modified model/SCHP/utils/__pycache__/transforms.cpython-39.pyc
Binary file not shown.

0 comments on commit fe6be79

Please sign in to comment.