Update app_gradio.py

app_gradio.py

@@ -25,6 +25,7 @@ from gifs_filter import filter
 import subprocess
 import uuid
 import tempfile
+import gc
 
 from huggingface_hub import snapshot_download
 
@@ -187,6 +188,7 @@ def generate_output(image, apply_filter, prompt: str, num_seeds: int = 3, lambda
         )
     except Exception as e:
         torch.cuda.empty_cache()  # Clear CUDA cache in case of failure
+        gc.collect()
         raise gr.Error(f"Video processing failed: {str(e)}") from e
     
     if apply_filter:
@@ -194,12 +196,17 @@ def generate_output(image, apply_filter, prompt: str, num_seeds: int = 3, lambda
             print("APPLYING FILTER")
             # Attempt to apply filtering
             filtered_gifs = filter(generated_gifs, temp_image_path)
+            torch.cuda.empty_cache()  # Clear CUDA cache in case of failure
+            gc.collect()
             return filtered_gifs, filtered_gifs
         except Exception as e:
             torch.cuda.empty_cache()  # Clear CUDA cache in case of failure
+            gc.collect()
             raise gr.Error(f"Filtering failed: {str(e)}") from e
     else:
         print("NOT APPLYING FILTER")
+        torch.cuda.empty_cache()  # Clear CUDA cache in case of failure
+        gc.collect()
         return generated_gifs, generated_gifs
 
 def generate_output_from_sketchpad(image, apply_filter, prompt: str, num_seeds: int = 3, lambda_value: float = 0.5, progress=gr.Progress(track_tqdm=True)):