Spaces:

geyongtao
/

HumanWild

Running on Zero

geyongtao commited on Jul 12

Commit

09954f9

•

1 Parent(s): 6eed647

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,7 +20,7 @@ from mini_dust3r.utils.misc import (
 import os
 from mini_dust3r.model import load_model
-from catmlp_dpt_head import Cat_MLP_LocalFeatures_DPT_Pts3d
 DEVICE = "cuda" if torch.cuda.is_available() else "CPU"

 import os
 from mini_dust3r.model import load_model
+from catmlp_dpt_head import Cat_MLP_LocalFeatures_DPT_Pts3d, postprocess
 DEVICE = "cuda" if torch.cuda.is_available() else "CPU"