fix: using zerogpu

app.py

@@ -1,11 +1,8 @@
 import argparse
 from pathlib import Path
-import spaces
 from imcui.ui.app_class import ImageMatchingApp
 
-
-@spaces.GPU(duration=60)
-def launch_app():
+if __name__ == "__main__":
     parser = argparse.ArgumentParser()
     parser.add_argument(
         "--server_name",
@@ -32,7 +29,3 @@ def launch_app():
         config=args.config,
         example_data_root=Path("imcui/datasets"),
     ).run()
-
-
-if __name__ == "__main__":
-    launch_app()

imcui/ui/utils.py

@@ -7,6 +7,7 @@ from itertools import combinations
 from pathlib import Path
 from typing import Any, Callable, Dict, List, Optional, Tuple, Union
 from datasets import load_dataset
+import spaces
 
 import cv2
 import gradio as gr
@@ -48,7 +49,7 @@ GRADIO_VERSION = gr.__version__.split(".")[0]
 MATCHER_ZOO = None
 
 
-model_cache = ModelCache(max_gpu_mem = 48e9, max_cpu_mem=48e9)
+model_cache = ModelCache(max_gpu_mem=48e9, max_cpu_mem=48e9)
 
 
 def load_config(config_name: str) -> Dict[str, Any]:
@@ -150,6 +151,7 @@ def download_example_images(repo_id, output_dir):
     logger.info(f"Images saved to {output_dir} successfully.")
     return Path(output_dir)
 
+
 def gen_examples(data_root: Path):
     random.seed(1)
     example_algos = [
@@ -826,6 +828,7 @@ def generate_fake_outputs(
     )
 
 
+@spaces.GPU(duration=60)
 def run_matching(
     image0: np.ndarray,
     image1: np.ndarray,