Spaces:
Running
Running
evanzyfan commited on
Commit ·
7a6f98c
1
Parent(s): a434439
update app.py
Browse files
app.py
CHANGED
|
@@ -22,7 +22,7 @@ from huggingface_hub import CommitScheduler, snapshot_download
|
|
| 22 |
# ============================================================================
|
| 23 |
|
| 24 |
DATA_DIR = os.environ.get("DATA_DIR", "./data")
|
| 25 |
-
OUTPUT_DIR = os.environ.get("OUTPUT_DIR", "./
|
| 26 |
NUM_GROUPS = int(os.environ.get("NUM_GROUPS", "10"))
|
| 27 |
RESULTS_REPO_ID = os.environ.get("RESULTS_REPO_ID", "")
|
| 28 |
DATA_REPO_ID = os.environ.get("DATA_REPO_ID", "MovieBench/moviebench-data2")
|
|
@@ -44,12 +44,13 @@ Path(OUTPUT_DIR).mkdir(parents=True, exist_ok=True)
|
|
| 44 |
|
| 45 |
scheduler: Optional[CommitScheduler] = None
|
| 46 |
if RESULTS_REPO_ID:
|
|
|
|
| 47 |
scheduler = CommitScheduler(
|
| 48 |
repo_id=RESULTS_REPO_ID,
|
| 49 |
repo_type="dataset",
|
| 50 |
folder_path=OUTPUT_DIR,
|
| 51 |
every=3,
|
| 52 |
-
path_in_repo="
|
| 53 |
token=HF_TOKEN,
|
| 54 |
)
|
| 55 |
|
|
|
|
| 22 |
# ============================================================================
|
| 23 |
|
| 24 |
DATA_DIR = os.environ.get("DATA_DIR", "./data")
|
| 25 |
+
OUTPUT_DIR = os.environ.get("OUTPUT_DIR", "./results")
|
| 26 |
NUM_GROUPS = int(os.environ.get("NUM_GROUPS", "10"))
|
| 27 |
RESULTS_REPO_ID = os.environ.get("RESULTS_REPO_ID", "")
|
| 28 |
DATA_REPO_ID = os.environ.get("DATA_REPO_ID", "MovieBench/moviebench-data2")
|
|
|
|
| 44 |
|
| 45 |
scheduler: Optional[CommitScheduler] = None
|
| 46 |
if RESULTS_REPO_ID:
|
| 47 |
+
print(f"Using scheduler for {RESULTS_REPO_ID} ...")
|
| 48 |
scheduler = CommitScheduler(
|
| 49 |
repo_id=RESULTS_REPO_ID,
|
| 50 |
repo_type="dataset",
|
| 51 |
folder_path=OUTPUT_DIR,
|
| 52 |
every=3,
|
| 53 |
+
path_in_repo="rk_pref",
|
| 54 |
token=HF_TOKEN,
|
| 55 |
)
|
| 56 |
|