Update run/gradio_ootd.py

run/gradio_ootd.py

@@ -76,7 +76,7 @@ def process_hd(vton_img, garm_img, n_samples, n_steps, image_scale, seed):
     return images
 
 @spaces.GPU
-def process_dc(vton_img, garm_img, category, n_samples, n_steps, image_scale, seed):
+def process_dc(vton_img, garm_img, category):
     model_type = 'dc'
     if category == 'Upper-body':
         category = 0
@@ -109,10 +109,10 @@ def process_dc(vton_img, garm_img, category, n_samples, n_steps, image_scale, se
             image_vton=masked_vton_img,
             mask=mask,
             image_ori=vton_img,
-            num_samples=n_samples,
-            num_steps=n_steps,
-            image_scale=image_scale,
-            seed=seed,
+            num_samples=1,
+            num_steps=20,
+            image_scale=2.0,
+            seed=-1,
         )
 
     return images
@@ -282,7 +282,7 @@ with block:
         # image_scale_dc = gr.Slider(label="Guidance scale", minimum=1.0, maximum=5.0, value=2.0, step=0.1)
         # seed_dc = gr.Slider(label="Seed", minimum=-1, maximum=2147483647, step=1, value=-1)
         
-    ips_dc = [vton_img_dc, garm_img_dc, category_dc, 1, 20, 2.0, -1]
+    ips_dc = [vton_img_dc, garm_img_dc, category_dc]
     run_button_dc.click(fn=process_dc, inputs=ips_dc, outputs=[result_gallery_dc])