Update app.py

app.py

@@ -23,7 +23,7 @@ task_name = 'depth'
 pipe_g, pipe_d = load_models(task_name, device)
 
 # Preprocess the video to adjust resolution and frame rate
-def preprocess_video(video_path, target_fps=24, max_resolution=(256, 256)):
+def preprocess_video(video_path, target_fps=24, max_resolution=(512, 512)):
     """Preprocess the video to resize and adjust its frame rate."""
     video = mp.VideoFileClip(video_path)
 
@@ -38,7 +38,7 @@ def preprocess_video(video_path, target_fps=24, max_resolution=(256, 256)):
     return video
 
 # Process a batch of frames through the depth model
-def process_frames_batch(frames_batch, seed=0, target_size=(256, 256)):
+def process_frames_batch(frames_batch, seed=0, target_size=(512, 512)):
     """Process a batch of frames and return depth maps."""
     try:
         torch.cuda.empty_cache()  # Clear GPU cache
@@ -64,7 +64,7 @@ def process_video(video_path, fps=0, seed=0, batch_size=4):
         start_time = time.time()
 
         # Preprocess the video
-        video = preprocess_video(video_path, target_fps=fps, max_resolution=(256, 256))
+        video = preprocess_video(video_path, target_fps=fps, max_resolution=(512, 512))
 
         # Use original video FPS if not specified
         if fps == 0:
@@ -149,8 +149,8 @@ def process_video(video_path, fps=0, seed=0, batch_size=4):
         yield None, None, None, f"Error processing video: {e}"
 
     finally:
-        # Clean up temporary directory
-        pass  # Remove if you decide to delete temp_dir
+        # Clean up temporary directory if necessary
+        pass
 
 # Wrapper function with error handling
 def process_wrapper(video, fps=0, seed=0, batch_size=4):