Update app.py

app.py

@@ -3,6 +3,7 @@ import cv2
 import torch
 import numpy as np
 import gradio as gr
+import spaces
 from huggingface_hub import hf_hub_download
 from depth_anything_v2.dpt import DepthAnythingV2
 
@@ -211,6 +212,7 @@ def initialize_model():
 # Initialize model at startup
 MODEL = initialize_model()
 
+@spaces.GPU
 def process_image(input_image):
     """
     Process the input image and return depth map and normal map
@@ -232,7 +234,7 @@ def process_image(input_image):
         # Normalize depth for visualization (0-255)
         depth_normalized = ((depth - depth.min()) / (depth.max() - depth.min()) * 255).astype(np.uint8)
     
-    # Move model back to CPU after processing
+    # Move model back to CPU
     MODEL.to('cpu')
     
     # Get intensity map
@@ -251,6 +253,7 @@ def process_image(input_image):
     
     return depth_normalized, normal_map
 
+@spaces.GPU
 def gradio_interface(input_img):
     try:
         depth_raw, normal = process_image(input_img)