download facexlib parsing_bisenet.pth at launch

app.py

@@ -17,7 +17,7 @@ from preprocessor import FaceProcessor
 
 from huggingface_hub import snapshot_download
 
-print("Loading model, please wait...")
+print("Downloading models, please wait...")
 
 snapshot_download("Wan-AI/Wan2.1-T2V-14B", local_dir="checkpoints/base_model/")
 
@@ -27,6 +27,16 @@ snapshot_download(
 )
 snapshot_download("BowenXue/Stand-In", local_dir="checkpoints/Stand-In/")
 
+import urllib.request
+
+weights_path = "/usr/local/lib/python3.10/site-packages/facexlib/weights/parsing_bisenet.pth"
+url = "https://github.com/xinntao/facexlib/releases/download/v0.2.0/parsing_bisenet.pth"
+
+if not os.path.exists(weights_path):
+    os.makedirs(os.path.dirname(weights_path), exist_ok=True)
+    print(f"Downloading model to {weights_path} ...")
+    urllib.request.urlretrieve(url, weights_path)
+
 try:
     ANTELOPEV2_PATH = "checkpoints/antelopev2"
     BASE_MODEL_PATH = "checkpoints/base_model/"
@@ -93,8 +103,8 @@ def generate_video(
 
     if is_shared_ui: 
         num_steps = 10
-        quality = 5
-        fps = 12
+        quality = 6
+        #fps = 12
         
     
     print("Generating video...")
@@ -186,7 +196,7 @@ with gr.Blocks() as demo:
                     info="More steps may improve details but will take longer to generate.",
                 )
                 output_fps = gr.Slider(
-                    label="Video FPS", minimum=10, maximum=30, step=1, value=12,
+                    label="Video FPS", minimum=10, maximum=30, step=1, value=25,
                     interactive = False if is_shared_ui else True,
                 )
                 output_quality = gr.Slider(