paper_qa / utils.py
chansung's picture
major update
7e4123a
raw
history blame
532 Bytes
import datasets
import pandas as pd
from huggingface_hub import create_repo
from huggingface_hub.utils import HfHubHTTPError
def push_to_hf_hub(
ds, repo_id, hf_token, append=True
):
exist = False
try:
create_repo(repo_id, repo_type="dataset", token=hf_token)
except HfHubHTTPError as e:
exist = True
if exist and append:
existing_ds = datasets.load_dataset(repo_id)
ds = datasets.concatenate_datasets([existing_ds['train'], ds])
ds.push_to_hub(repo_id, token=hf_token)