Update app_gradio.py

app_gradio.py

@@ -155,7 +155,7 @@ def process_video(num_frames, num_seeds, generator, exp_dir, load_name, caption,
 
     return gifs
 
-def generate_output(image, prompt: str, num_seeds: int = 3, lambda_value: float = 0.5, apply_filer: bool = True, progress=gr.Progress(track_tqdm=True)) -> List[str]:
+def generate_output(image, prompt: str, num_seeds: int = 3, lambda_value: float = 0.5, progress=gr.Progress(track_tqdm=True)) -> List[str]:
     """Main function to generate output GIFs"""
     exp_dir = "static/app_tmp"
     os.makedirs(exp_dir, exist_ok=True)
@@ -175,19 +175,15 @@ def generate_output(image, prompt: str, num_seeds: int = 3, lambda_value: float
         lambda_=1 - lambda_value
     )
 
-    if apply_filter is True:
-        # Apply filtering (assuming filter function is imported)
-        print("APPLYING FILTERS")
-        filtered_gifs = filter(generated_gifs, temp_image_path)
-        results = filtered_gifs
-    else:
-        results = generated_gifs
+    # Apply filtering (assuming filter function is imported)
+    print("APPLYING FILTERS")
+    filtered_gifs = filter(generated_gifs, temp_image_path)   
     
-    return results
+    return filtered_gifs
 
-def generate_output_from_sketchpad(image, prompt: str, num_seeds: int = 3, lambda_value: float = 0.5, apply_filer: bool = True, progress=gr.Progress(track_tqdm=True)):
+def generate_output_from_sketchpad(image, prompt: str, num_seeds: int = 3, lambda_value: float = 0.5, progress=gr.Progress(track_tqdm=True)):
     image = image['composite']
-    results = generate_output(image, prompt, num_seeds, lambda_value, apply_filter)
+    results = generate_output(image, prompt, num_seeds, lambda_value)
     return results
 
 css=""" """
@@ -308,11 +304,6 @@ with gr.Blocks(css=css) as demo:
                         step=0.1, 
                         label="Motion Strength"
                     )
-                    apply_filter = gr.Checkbox(
-                        label="Apply GIFs Filters",
-                        info="If checked, bad/off results will be filtered",
-                        value=True
-                    )
                     
                     sketchpad_generate_btn = gr.Button(
                         "Generate Animation",
@@ -368,7 +359,6 @@ with gr.Blocks(css=css) as demo:
                 draw_motion_prompt,
                 draw_num_seeds,
                 draw_lambda_,
-                apply_filter
             ],
             outputs=output_gallery
     )