Update app.py
Browse files
app.py
CHANGED
|
@@ -123,7 +123,7 @@ automasker = AutoMasker(
|
|
| 123 |
device='cuda',
|
| 124 |
)
|
| 125 |
|
| 126 |
-
@spaces.GPU(duration=120)
|
| 127 |
def submit_function(
|
| 128 |
person_image,
|
| 129 |
cloth_image,
|
|
@@ -207,35 +207,6 @@ def person_example_fn(image_path):
|
|
| 207 |
return image_path
|
| 208 |
|
| 209 |
HEADER = """
|
| 210 |
-
<h1 style="text-align: center;"> 馃悎 CatVTON: Concatenation Is All You Need for Virtual Try-On with Diffusion Models </h1>
|
| 211 |
-
<div style="display: flex; justify-content: center; align-items: center;">
|
| 212 |
-
<a href="http://arxiv.org/abs/2407.15886" style="margin: 0 2px;">
|
| 213 |
-
<img src='https://img.shields.io/badge/arXiv-2407.15886-red?style=flat&logo=arXiv&logoColor=red' alt='arxiv'>
|
| 214 |
-
</a>
|
| 215 |
-
<a href='https://huggingface.co/zhengchong/CatVTON' style="margin: 0 2px;">
|
| 216 |
-
<img src='https://img.shields.io/badge/Hugging Face-ckpts-orange?style=flat&logo=HuggingFace&logoColor=orange' alt='huggingface'>
|
| 217 |
-
</a>
|
| 218 |
-
<a href="https://github.com/Zheng-Chong/CatVTON" style="margin: 0 2px;">
|
| 219 |
-
<img src='https://img.shields.io/badge/GitHub-Repo-blue?style=flat&logo=GitHub' alt='GitHub'>
|
| 220 |
-
</a>
|
| 221 |
-
<a href="http://120.76.142.206:8888" style="margin: 0 2px;">
|
| 222 |
-
<img src='https://img.shields.io/badge/Demo-Gradio-gold?style=flat&logo=Gradio&logoColor=red' alt='Demo'>
|
| 223 |
-
</a>
|
| 224 |
-
<a href="https://huggingface.co/spaces/zhengchong/CatVTON" style="margin: 0 2px;">
|
| 225 |
-
<img src='https://img.shields.io/badge/Space-ZeroGPU-orange?style=flat&logo=Gradio&logoColor=red' alt='Demo'>
|
| 226 |
-
</a>
|
| 227 |
-
<a href='https://zheng-chong.github.io/CatVTON/' style="margin: 0 2px;">
|
| 228 |
-
<img src='https://img.shields.io/badge/Webpage-Project-silver?style=flat&logo=&logoColor=orange' alt='webpage'>
|
| 229 |
-
</a>
|
| 230 |
-
<a href="https://github.com/Zheng-Chong/CatVTON/LICENCE" style="margin: 0 2px;">
|
| 231 |
-
<img src='https://img.shields.io/badge/License-CC BY--NC--SA--4.0-lightgreen?style=flat&logo=Lisence' alt='License'>
|
| 232 |
-
</a>
|
| 233 |
-
</div>
|
| 234 |
-
<br>
|
| 235 |
-
路 This demo and our weights are only for Non-commercial Use. <br>
|
| 236 |
-
路 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>
|
| 237 |
-
路 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>
|
| 238 |
-
路 SafetyChecker is set to filter NSFW content, but it may block normal results too. Please adjust the <span>`seed`</span> for normal outcomes.<br>
|
| 239 |
"""
|
| 240 |
|
| 241 |
def app_gradio():
|
|
|
|
| 123 |
device='cuda',
|
| 124 |
)
|
| 125 |
|
| 126 |
+
# @spaces.GPU(duration=120)
|
| 127 |
def submit_function(
|
| 128 |
person_image,
|
| 129 |
cloth_image,
|
|
|
|
| 207 |
return image_path
|
| 208 |
|
| 209 |
HEADER = """
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 210 |
"""
|
| 211 |
|
| 212 |
def app_gradio():
|