Shad0ws commited on
Commit
021f341
verified
1 Parent(s): 32b0bc1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -30
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():