Fix

app.py

@@ -94,7 +94,7 @@ def predict_depth_v1(image, model_name):
 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)
-    model = depth_anything_v2_models[model_name]
+    model = depth_anything_v2_models[model_name].cuda()
     return model.infer_image(image)
 
 
@@ -115,6 +115,7 @@ def compute_depth_map_v1(image, model_select):
 
 
 @spaces.GPU
+@torch.no_grad()
 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)
@@ -142,10 +143,7 @@ with gr.Blocks(css=css) as demo:
     example_files = os.listdir('assets/examples')
     example_files.sort()
     example_files = [os.path.join('assets/examples', filename) for filename in example_files]
-    examples = gr.Examples(
-        examples=example_files, inputs=[input_image, model_select_v1, model_select_v2],
-        outputs=[depth_image_slider], fn=on_submit, cache_examples="lazy",
-    )
+    examples = gr.Examples(examples=example_files, inputs=[input_image])
 
 
 if __name__ == '__main__':