update

README.md

@@ -1,6 +1,6 @@
 ---
-title: Kolors Tryon
-emoji: 🖼
+title: Kolors Visual Tryon
+emoji: 👕
 colorFrom: purple
 colorTo: gray
 sdk: gradio

app.py

@@ -5,15 +5,18 @@ from PIL import Image
 import gradio as gr
 import numpy as np
 import random
+import spaces
 
-example_path = os.path.join(os.path.dirname(__file__), 'assets')
-
-MAX_SEED = 999999
 
+@spaces.GPU
 def start_tryon(imgs, garm_img, garment_des, seed):
     
     return None
 
+MAX_SEED = 999999
+
+example_path = os.path.join(os.path.dirname(__file__), 'assets')
+
 garm_list = os.listdir(os.path.join(example_path,"cloth"))
 garm_list_path = [os.path.join(example_path,"cloth",garm) for garm in garm_list]
 
@@ -63,8 +66,14 @@ with gr.Blocks(css=css) as Tryon:
 
     with gr.Column():
         with gr.Accordion(label="Advanced Settings", open=False):
-            with gr.Row():
-                seed = gr.Number(label="Seed", minimum=-1, maximum=2147483647, step=1, value=None)
+            seed = gr.Slider(
+                    label="Seed",
+                    minimum=0,
+                    maximum=MAX_SEED,
+                    step=1,
+                    value=0,
+                )
+            randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
 
     try_button.click(fn=start_tryon, inputs=[imgs, garm_img, seed], outputs=[image_out], api_name='tryon')
 

assets/title.md

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ee7fff5b3874ec80927627cdec9b2e5773cf257952b81fd280c4458993dc1894
-size 598
+oid sha256:79f28d9f0cc8ac8b370ab4bf945a9450dfb6c7febdf3c56d26b1cecd4d386632
+size 604