clean up tmp dir after each run

app.py

@@ -6,6 +6,7 @@ from typing import Any, Mapping, Sequence, Union
 import gradio as gr
 import numpy as np
 import spaces
+import time
 import torch
 import yaml
 from huggingface_hub import hf_hub_download
@@ -208,9 +209,9 @@ def clear_tmp_directory():
                     full_path = os.path.join(root, file)
                     try:
                         os.remove(full_path)
-                        print(f"Deleted: {full_path}")
+                        print("Deleted an image.")
                     except Exception as e:
-                        print(f"Failed to delete {full_path}: {e}")
+                        print("Failed to delete an image.")
 
 
 add_comfyui_directory_to_sys_path()
@@ -219,6 +220,7 @@ add_extra_model_paths()
 
 @spaces.GPU(duration=60)
 def advance_blur(input_image):
+    start_time = time.time()
     with torch.inference_mode():
         loaded_input_image = loadimage.load_image(
             image=input_image,
@@ -282,6 +284,11 @@ def advance_blur(input_image):
         # Clean up the /tmp directory
         clear_tmp_directory()
 
+        # Log the time taken for the process
+        end_time = time.time()
+        elapsed_time = end_time - start_time
+        print(f"Processing time: {elapsed_time:.2f} seconds")
+
         return outpath