daidedou commited on
Commit
38ea30e
·
1 Parent(s): 3ccb54b

Remove pykeops for space ZeroGPU demo.

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -160,6 +160,7 @@ def build_outputs(surf_a: Surface, surf_b: Surface, cmap_a: np.ndarray, p2p: np.
160
  @spaces.GPU
161
  def init_clicked(mesh1_path, mesh2_path,
162
  lambda_val, zoomout_val, time_val, nloop_val, sds_val, proper_val):
 
163
  cfg.deepfeat_conf.fmap.lambda_ = lambda_val
164
  cfg.sds_conf.zoomout = zoomout_val
165
  cfg.deepfeat_conf.fmap.diffusion.time = time_val
@@ -269,5 +270,6 @@ if __name__ == "__main__":
269
  args = parser.parse_args()
270
  cfg = OmegaConf.load(args.config)
271
  matcher = zero_shot.Matcher(cfg)
 
272
  datadicts = None
273
  demo.launch(share=args.share)
 
160
  @spaces.GPU
161
  def init_clicked(mesh1_path, mesh2_path,
162
  lambda_val, zoomout_val, time_val, nloop_val, sds_val, proper_val):
163
+ print("inside init")
164
  cfg.deepfeat_conf.fmap.lambda_ = lambda_val
165
  cfg.sds_conf.zoomout = zoomout_val
166
  cfg.deepfeat_conf.fmap.diffusion.time = time_val
 
270
  args = parser.parse_args()
271
  cfg = OmegaConf.load(args.config)
272
  matcher = zero_shot.Matcher(cfg)
273
+ os.makedirs("tmp", exist_ok=True)
274
  datadicts = None
275
  demo.launch(share=args.share)