Update app.py

app.py

@@ -8,7 +8,7 @@ import imageio as imageio
 import numpy as np
 import spaces
 import torch as torch
-from PIL import Image
+from PIL import Image, ImageFilter
 from tqdm import tqdm
 from pathlib import Path
 import gradio
@@ -23,9 +23,17 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 if "HF_TOKEN_LOGIN" in os.environ:
     login(token=os.environ["HF_TOKEN_LOGIN"])
 
+def apply_gaussian_blur(image, radius=0.75):
+    """Apply Gaussian blur to PIL Image with specified radius"""
+    return image.filter(ImageFilter.GaussianBlur(radius=radius))
+
 def infer(path_input, seed=None):
     name_base, name_ext = os.path.splitext(os.path.basename(path_input))
     _, output_d = lotus(path_input, 'depth', seed, device)
+    
+    # Apply Gaussian blur with 0.75 radius
+    output_d = apply_gaussian_blur(output_d, radius=0.75)
+    
     if not os.path.exists("files/output"):
         os.makedirs("files/output")
     d_save_path = os.path.join("files/output", f"{name_base}_d{name_ext}")
@@ -34,11 +42,25 @@ def infer(path_input, seed=None):
 
 def infer_video(path_input, seed=None):
     _, frames_d, fps = lotus_video(path_input, 'depth', seed, device)
+    
+    # Apply Gaussian blur to each frame
+    blurred_frames = []
+    for frame in frames_d:
+        # Convert numpy array to PIL Image if needed
+        if isinstance(frame, np.ndarray):
+            frame_pil = Image.fromarray(frame)
+        else:
+            frame_pil = frame
+        
+        # Apply blur and convert back to numpy array
+        blurred_frame = apply_gaussian_blur(frame_pil, radius=0.75)
+        blurred_frames.append(np.array(blurred_frame))
+    
     if not os.path.exists("files/output"):
         os.makedirs("files/output")
     name_base, _ = os.path.splitext(os.path.basename(path_input))
     d_save_path = os.path.join("files/output", f"{name_base}_d.mp4")
-    imageio.mimsave(d_save_path, frames_d, fps=fps)
+    imageio.mimsave(d_save_path, blurred_frames, fps=fps)
     return d_save_path
 
 def run_demo_server():