Relocate GPU

app.py

@@ -84,7 +84,6 @@ def get_v2_model(model_name):
     return model
 
 
-@spaces.GPU
 def predict_depth_v1(image, model_name):
     if model_name not in depth_anything_v1_pipelines:
         depth_anything_v1_pipelines[model_name] = get_v1_pipe(model_name)
@@ -92,7 +91,6 @@ def predict_depth_v1(image, model_name):
     return pipe(image)
 
 
-@spaces.GPU
 def predict_depth_v2(image, model_name):
     if model_name not in depth_anything_v2_models:
         depth_anything_v2_models[model_name] = get_v2_model(model_name)
@@ -116,6 +114,7 @@ def compute_depth_map_v1(image, model_select):
     return colored_depth
 
 
+@spaces.GPU
 def on_submit(image, model_v1_select, model_v2_select):
     logger.info(f"Computing depth for V1 model: {model_v1_select} and V2 model: {model_v2_select}")
     colored_depth_v1 = compute_depth_map_v1(image, model_v1_select)