chaoxu commited on
Commit
dca5722
โ€ข
1 Parent(s): 3348d39

elev estimate update

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ code_dir = snapshot_download("One-2-3-45/code") if not is_local_run else "../cod
8
 
9
  sys.path.append(code_dir)
10
 
11
- elev_est_dir = os.path.abspath(os.path.join(code_dir, "one2345_elev_est"))
12
  sys.path.append(elev_est_dir)
13
 
14
  if not is_local_run:
@@ -36,7 +36,7 @@ import tempfile
36
  from zero123_utils import init_model, predict_stage1_gradio, zero123_infer
37
  from sam_utils import sam_init, sam_out_nosave
38
  from utils import image_preprocess_nosave, gen_poses
39
- from one2345_elev_est.tools.estimate_wild_imgs import estimate_elev
40
  from rembg import remove
41
 
42
  _GPU_INDEX = 0
 
8
 
9
  sys.path.append(code_dir)
10
 
11
+ elev_est_dir = os.path.abspath(os.path.join(code_dir, "elevation_estimate"))
12
  sys.path.append(elev_est_dir)
13
 
14
  if not is_local_run:
 
36
  from zero123_utils import init_model, predict_stage1_gradio, zero123_infer
37
  from sam_utils import sam_init, sam_out_nosave
38
  from utils import image_preprocess_nosave, gen_poses
39
+ from elevation_estimate.estimate_wild_imgs import estimate_elev
40
  from rembg import remove
41
 
42
  _GPU_INDEX = 0