Update app.py

app.py

@@ -57,9 +57,6 @@ def extract_frames_with_labels(video_path, base_output_dir="frames"):
 
 # Define a function to run your script with selected inputs
 def run_xportrait(source_image, driving_video, seed, uc_scale, best_frame, out_frames, num_mix, ddim_steps):
-    # Check if the model weights are in place
-    if not os.path.exists(resume_dir):
-        return "Model weights not found in checkpoint directory. Please download them first."
 
     # Create a unique output directory name based on current date and time
     output_dir_base = "outputs"
@@ -115,7 +112,7 @@ with gr.Blocks() as demo:
                         best_frame = gr.Number(value=36, label="Best Frame")
                         out_frames = gr.Number(value=-1, label="Out Frames")
                     with gr.Accordion("Driving video Frames"):
-                        driving_frames = gr.Gallery(show_label=True)
+                        driving_frames = gr.Gallery(show_label=True, columns=6, height=512)
                 with gr.Row():
                     num_mix = gr.Number(value=4, label="Number of Mix")
                     ddim_steps = gr.Number(value=30, label="DDIM Steps")