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():
|