Spaces:

theaiinstitute
/

theia

Running on Zero

bmay commited on Aug 1, 2024

Commit

6d5e85a

verified ·

1 Parent(s): 1cd3984

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ def run_theia(image):
         "facebook/sam-vit-huge",
         "LiheYoung/depth-anything-large-hf",
     ]
-    feature_means, feature_vars = load_feature_stats(target_model_names, stat_file_root="../../../feature_stats")
     mask_generator, sam_model = prepare_mask_generator(device)
     depth_anything_model_name = "LiheYoung/depth-anything-large-hf"

         "facebook/sam-vit-huge",
         "LiheYoung/depth-anything-large-hf",
     ]
+    feature_means, feature_vars = load_feature_stats(target_model_names, stat_file_root="feature_stats")
     mask_generator, sam_model = prepare_mask_generator(device)
     depth_anything_model_name = "LiheYoung/depth-anything-large-hf"