sumitasthaai commited on
Commit
d85073c
·
unverified ·
1 Parent(s): f4bd3fb

Use python huggingface_hub API instead of CLI for downloading DBs

Browse files
Files changed (1) hide show
  1. start.sh +10 -2
start.sh CHANGED
@@ -4,12 +4,20 @@ set -e
4
  # Download SQLite databases from private HF dataset using secret token
5
  if [ ! -f "voter_db_new.sqlite3" ]; then
6
  echo "Downloading voter_db_new.sqlite3..."
7
- huggingface-cli download rockerritesh/electionnepal-data voter_db_new.sqlite3 --repo-type dataset --local-dir . --token "$HF_TOKEN"
 
 
 
 
8
  fi
9
 
10
  if [ ! -f "voter_db_old.sqlite3" ]; then
11
  echo "Downloading voter_db_old.sqlite3..."
12
- huggingface-cli download rockerritesh/electionnepal-data voter_db_old.sqlite3 --repo-type dataset --local-dir . --token "$HF_TOKEN"
 
 
 
 
13
  fi
14
 
15
  echo "Starting Streamlit..."
 
4
  # Download SQLite databases from private HF dataset using secret token
5
  if [ ! -f "voter_db_new.sqlite3" ]; then
6
  echo "Downloading voter_db_new.sqlite3..."
7
+ python3 -c "
8
+ from huggingface_hub import hf_hub_download
9
+ import os
10
+ hf_hub_download('rockerritesh/electionnepal-data', 'voter_db_new.sqlite3', repo_type='dataset', token=os.environ['HF_TOKEN'], local_dir='.')
11
+ "
12
  fi
13
 
14
  if [ ! -f "voter_db_old.sqlite3" ]; then
15
  echo "Downloading voter_db_old.sqlite3..."
16
+ python3 -c "
17
+ from huggingface_hub import hf_hub_download
18
+ import os
19
+ hf_hub_download('rockerritesh/electionnepal-data', 'voter_db_old.sqlite3', repo_type='dataset', token=os.environ['HF_TOKEN'], local_dir='.')
20
+ "
21
  fi
22
 
23
  echo "Starting Streamlit..."