Eran Fainman commited on
Commit
ac48997
Β·
1 Parent(s): 827117f

Add application file

Browse files
Files changed (1) hide show
  1. app.py +13 -5
app.py CHANGED
@@ -7,17 +7,25 @@ st.set_page_config(page_title="RecTour2024 - Booking.com Review Ranking Challeng
7
  st.title("πŸ† RecTour2024 Leaderboard")
8
 
9
  # read files from HF
10
- RESULTS_REPO = "Booking-com/results"
 
 
11
  TOKEN = os.environ.get("HF_TOKEN")
12
  CACHE_PATH = os.getenv("HF_HOME", ".")
13
  EVAL_RESULTS_PATH = os.path.join(CACHE_PATH, "eval-results")
14
 
15
  API = hh.HfApi(token=TOKEN)
16
 
17
- hh.snapshot_download(
18
- repo_id=RESULTS_REPO, local_dir=EVAL_RESULTS_PATH, repo_type=None, tqdm_class=None, etag_timeout=30,
19
- token=TOKEN
20
- )
 
 
 
 
 
 
21
 
22
  df_results = pd.read_csv(os.path.join(EVAL_RESULTS_PATH, 'results.csv'))
23
  st.table(df_results)
 
7
  st.title("πŸ† RecTour2024 Leaderboard")
8
 
9
  # read files from HF
10
+ OWNER = "Booking-com"
11
+ REPO_ID = f"{OWNER}/streamlit-review-ranking-leaderboard"
12
+ RESULTS_REPO = f"{OWNER}/results"
13
  TOKEN = os.environ.get("HF_TOKEN")
14
  CACHE_PATH = os.getenv("HF_HOME", ".")
15
  EVAL_RESULTS_PATH = os.path.join(CACHE_PATH, "eval-results")
16
 
17
  API = hh.HfApi(token=TOKEN)
18
 
19
+ def restart_space():
20
+ API.restart_space(repo_id=REPO_ID)
21
+
22
+ try:
23
+ hh.snapshot_download(
24
+ repo_id=RESULTS_REPO, local_dir=EVAL_RESULTS_PATH, repo_type=None, tqdm_class=None, etag_timeout=30,
25
+ token=TOKEN
26
+ )
27
+ except:
28
+
29
 
30
  df_results = pd.read_csv(os.path.join(EVAL_RESULTS_PATH, 'results.csv'))
31
  st.table(df_results)