evanzyfan commited on
Commit
7a6f98c
·
1 Parent(s): a434439

update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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", "./results_pref")
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="final_results/rk_pref",
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