xiaozaa commited on
Commit
8eb9095
1 Parent(s): 0a93c94

revise app

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -11,7 +11,7 @@ def gradio_inference(
11
  num_steps=50,
12
  guidance_scale=30.0,
13
  seed=-1,
14
- size=(576,768)
15
  ):
16
  """Wrapper function for Gradio interface"""
17
  # Use temporary directory
@@ -55,6 +55,9 @@ def create_demo():
55
  gr.Markdown("""
56
  # CATVTON FLUX Virtual Try-On Demo
57
  Upload a model image, an agnostic mask, and a garment image to generate virtual try-on results.
 
 
 
58
  """)
59
 
60
  with gr.Column():
 
11
  num_steps=50,
12
  guidance_scale=30.0,
13
  seed=-1,
14
+ size=(768,1024)
15
  ):
16
  """Wrapper function for Gradio interface"""
17
  # Use temporary directory
 
55
  gr.Markdown("""
56
  # CATVTON FLUX Virtual Try-On Demo
57
  Upload a model image, an agnostic mask, and a garment image to generate virtual try-on results.
58
+
59
+ [![Hugging Face Spaces](https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Spaces-blue)](https://huggingface.co/xiaozaa/catvton-flux-alpha)
60
+ [![GitHub](https://img.shields.io/badge/github-%23121011.svg?style=for-the-badge&logo=github&logoColor=white)](https://github.com/nftblackmagic/catvton-flux)
61
  """)
62
 
63
  with gr.Column():