update: gpu support

app.py

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

requirements.txt

@@ -35,6 +35,7 @@ scikit-learn
 scipy
 seaborn
 shapely
+spaces
 tensorboardX==2.6.1
 torchmetrics==0.6.0
 torchvision==0.19.0