emirkisa commited on
Commit
8ce3369
·
verified ·
1 Parent(s): b1d4267

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -68,6 +68,12 @@ else:
68
 
69
  DAVIS_ROOT = Path(os.environ.get("DAVIS_ROOT", str(_local_root)))
70
 
 
 
 
 
 
 
71
  IMG_DIR = DAVIS_ROOT / "JPEGImages" / "480p"
72
  ANN_DIR = DAVIS_ROOT / "Annotations" / "480p"
73
  SETS_DIR = DAVIS_ROOT / "ImageSets"
@@ -89,12 +95,6 @@ def _cleanup_stale_tmp() -> None:
89
 
90
  _cleanup_stale_tmp()
91
 
92
- _REPO_ROOT = Path(__file__).resolve().parents[2]
93
- PROCESSED_DAVIS_DIR = Path(os.environ.get(
94
- "PROCESSED_DAVIS_DIR",
95
- str(_REPO_ROOT / "data" / "processed" / "DAVIS"),
96
- ))
97
-
98
  DAVIS_PALETTE = np.array([
99
  [ 0, 0, 0], [128, 0, 0], [ 0, 128, 0], [128, 128, 0],
100
  [ 0, 0, 128], [128, 0, 128], [ 0, 128, 128], [128, 128, 128],
 
68
 
69
  DAVIS_ROOT = Path(os.environ.get("DAVIS_ROOT", str(_local_root)))
70
 
71
+ if IS_HF_SPACE:
72
+ _processed_default = _hf_base / "DAVIS_processed"
73
+ else:
74
+ _processed_default = Path("/workspace/diffusion-research/data/processed/DAVIS")
75
+ PROCESSED_DAVIS_DIR = Path(os.environ.get("PROCESSED_DAVIS_DIR", str(_processed_default)))
76
+
77
  IMG_DIR = DAVIS_ROOT / "JPEGImages" / "480p"
78
  ANN_DIR = DAVIS_ROOT / "Annotations" / "480p"
79
  SETS_DIR = DAVIS_ROOT / "ImageSets"
 
95
 
96
  _cleanup_stale_tmp()
97
 
 
 
 
 
 
 
98
  DAVIS_PALETTE = np.array([
99
  [ 0, 0, 0], [128, 0, 0], [ 0, 128, 0], [128, 128, 0],
100
  [ 0, 0, 128], [128, 0, 128], [ 0, 128, 128], [128, 128, 128],