Update app.py

app.py

@@ -64,6 +64,20 @@ class WhiteTheme(Base):
 # Set device
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
+# Add the preprocess_video function to limit video resolution and frame rate
+def preprocess_video(video_path, target_fps=24, max_resolution=(640, 360)):
+    """Preprocess the video to reduce its resolution and frame rate."""
+    video = mp.VideoFileClip(video_path)
+    
+    # Resize video if it's larger than the target resolution
+    if video.size[0] > max_resolution[0] or video.size[1] > max_resolution[1]:
+        video = video.resize(newsize=max_resolution)
+    
+    # Limit FPS
+    video = video.set_fps(target_fps)
+    
+    return video
+
 def process_frame(frame, seed=0):
     """
     Process a single frame through the depth model.
@@ -92,7 +106,7 @@ def process_frame(frame, seed=0):
         return None
 
 @spaces.GPU
-def process_video(video_path, fps=0, seed=0, max_workers=2):  # Reduced max_workers
+def process_video(video_path, fps=0, seed=0, max_workers=2):
     """
     Process video to create depth map sequence and video.
     Maintains original resolution and framerate if fps=0.