Update gradio/vace.py

gradio/vace.py

@@ -17,6 +17,8 @@ snapshot_download(
     local_dir = "./models/Wan2.1-VACE-1.3B"
 )
 
+is_shared_ui = True if "fffiloni/Wan2.1-VACE-1.3B" in os.environ['SPACE_ID'] else False
+
 sys.path.insert(0, os.path.sep.join(os.path.realpath(__file__).split(os.path.sep)[:-2]))
 import wan
 from wan import WanVace, WanVaceMP
@@ -65,11 +67,21 @@ class VACEInference:
 
 
     def create_ui(self, *args, **kwargs):
-        gr.Markdown("""
-                    <div style="text-align: center; font-size: 24px; font-weight: bold; margin-bottom: 15px;">
-                        <a href="https://ali-vilab.github.io/VACE-Page/" style="text-decoration: none; color: inherit;">VACE-WAN Demo</a>
-                    </div>
-                    """)
+
+        gr.Markdown("# VACE-WAN 1.3B Demo")
+        gr.Markdown("All-in-One Video Creation and Editing")
+
+        gr.HTML("""
+            <div style="display:flex;column-gap:4px;">
+                <a href="https://ali-vilab.github.io/VACE-Page/">
+                    <img src='https://img.shields.io/badge/Project-Page-green'>
+                </a> 
+    			
+                <a href="https://huggingface.co/spaces/fffiloni/Wan2.1-VACE-1.3B?duplicate=true">
+    				<img src="https://huggingface.co/datasets/huggingface/badges/resolve/main/duplicate-this-space-sm.svg" alt="Duplicate this Space">
+    			</a>	
+            </div>
+        """)
         with gr.Row(variant='panel', equal_height=True):
             with gr.Column(scale=1, min_width=0):
                 self.src_video = gr.Video(
@@ -150,7 +162,7 @@ class VACEInference:
                         maximum=100,
                         step=1,
                         value=25,
-                        interactive=True)
+                        interactive=False if is_shared_ui else True)
                     self.context_scale = gr.Slider(
                         label='context_scale',
                         minimum=0.0,
@@ -174,13 +186,13 @@ class VACEInference:
             with gr.Row(equal_height=True):
                 self.output_height = gr.Textbox(
                     label='resolutions_height',
-                    # value=480,
-                    value=720,
+                    value=480,
+                    #value=720,
                     interactive=True)
                 self.output_width = gr.Textbox(
                     label='resolutions_width',
-                    # value=832,
-                    value=1280,
+                    value=832,
+                    #value=1280,
                     interactive=True)
                 self.frame_rate = gr.Textbox(
                     label='frame_rate',
@@ -269,7 +281,7 @@ if __name__ == '__main__':
     parser.add_argument('--root_path', dest='root_path', help='', default=None)
     parser.add_argument('--save_dir', dest='save_dir', help='', default='cache')
     parser.add_argument("--mp", action="store_true", help="Use Multi-GPUs",)
-    parser.add_argument("--model_name", type=str, default="vace-14B", choices=list(WAN_CONFIGS.keys()), help="The model name to run.")
+    parser.add_argument("--model_name", type=str, default="vace-1.3B", choices=list(WAN_CONFIGS.keys()), help="The model name to run.")
     parser.add_argument("--ulysses_size", type=int, default=1, help="The size of the ulysses parallelism in DiT.")
     parser.add_argument("--ring_size", type=int, default=1, help="The size of the ring attention parallelism in DiT.")
     parser.add_argument(