Spaces:

lsxi77777
/

MINIMA

Running on Zero

lsxi77777 commited on Jan 17

Commit

89426d4

1 Parent(s): ea17c91

fix on gpu

Files changed (1) hide show

ui/app_class.py CHANGED Viewed

@@ -6,7 +6,7 @@ import gradio as gr
 import numpy as np
 from easydict import EasyDict as edict
 from omegaconf import OmegaConf
 sys.path.append(str(Path(__file__).parents[1]))
 from ui.sfm import SfmEngine
@@ -83,7 +83,7 @@ a:hover {
 }
 """
 class ImageMatchingApp:
     def __init__(self, server_name="0.0.0.0", server_port=7860, **kwargs):
         self.server_name = server_name

 import numpy as np
 from easydict import EasyDict as edict
 from omegaconf import OmegaConf
+import spaces
 sys.path.append(str(Path(__file__).parents[1]))
 from ui.sfm import SfmEngine
 }
 """
+@spaces.GPU
 class ImageMatchingApp:
     def __init__(self, server_name="0.0.0.0", server_port=7860, **kwargs):
         self.server_name = server_name