Update app.py

app.py

@@ -1,4 +1,3 @@
-import spaces  # Import the spaces module for ZeroGPU
 import gradio as gr
 import torch
 import os
@@ -9,8 +8,9 @@ import ffmpeg
 import numpy as np
 from PIL import Image
 import moviepy.editor as mp
-from infer import lotus, load_models
+from infer import lotus, load_models, pipe_g, pipe_d  # Import the global models
 import logging
+import spaces  # Import the spaces module for ZeroGPU
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
@@ -19,9 +19,15 @@ logger = logging.getLogger(__name__)
 # Device will be set inside GPU-decorated functions
 device = 'cuda'  # Use 'cuda' as placeholder
 
-# Load model names and dtype
+# Load models once inside a GPU context
 task_name = 'depth'
-model_g, model_d, dtype = load_models(task_name, device)
+
+@spaces.GPU
+def initialize_models():
+    load_models(task_name, device)
+
+# Call the function to load models
+initialize_models()
 
 # Preprocess the video to adjust frame rate
 def preprocess_video(video_path, target_fps=24):
@@ -58,7 +64,7 @@ def process_frame(frame, seed=0, target_size=(512, 512)):
         input_image = resize_and_pad(image, target_size)
 
         # Run inference
-        depth_map = lotus(input_image, 'depth', seed, device, model_g, model_d, dtype)
+        depth_map = lotus(input_image, 'depth', seed, device)
 
         # Crop the output depth map back to original image size
         width, height = image.size