Spaces:
Sleeping
Sleeping
daidedou
commited on
Commit
·
38ea30e
1
Parent(s):
3ccb54b
Remove pykeops for space ZeroGPU demo.
Browse files
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)
|