fix bug in cuda init

app.py

@@ -5,12 +5,6 @@ import os
 from pathlib import Path
 from typing import Optional
 
-import cv2
-import gradio as gr
-import numpy as np
-import torch
-from huggingface_hub import hf_hub_download
-
 
 def _ensure_localhost_no_proxy() -> None:
     hosts = ["127.0.0.1", "localhost", "::1"]
@@ -28,6 +22,19 @@ def _ensure_localhost_no_proxy() -> None:
 
 _ensure_localhost_no_proxy()
 
+try:
+    import spaces
+
+    GPU = spaces.GPU
+except ImportError:
+    GPU = lambda fn: fn
+
+import cv2
+import gradio as gr
+import numpy as np
+import torch
+from huggingface_hub import hf_hub_download
+
 from InfiniDepth.utils.hf_demo_utils import (
     DemoArtifacts,
     ensure_session_output_dir,
@@ -48,13 +55,6 @@ from InfiniDepth.utils.io_utils import depth_to_disparity
 from InfiniDepth.utils.model_utils import build_model
 from InfiniDepth.utils.sampling_utils import SAMPLING_METHODS
 
-try:
-    import spaces
-
-    GPU = spaces.GPU
-except ImportError:
-    GPU = lambda fn: fn
-
 
 APP_ROOT = Path(__file__).resolve().parent
 EXAMPLES_DIR = APP_ROOT / "example_data"