File size: 1,201 Bytes
2063a08 715553f 2063a08 c7bbc1c 2063a08 63a76d0 2063a08 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
import argparse
import os
from huggingface_hub import HfApi
TOKEN = os.environ.get("DATACOMP_TOKEN")
api = HfApi(token=TOKEN)
parser = argparse.ArgumentParser()
parser.add_argument(
"--out_dir",
default="teaching_arithmetic/out/",
type=str,
required=False,
help="Path to the output directory.",
)
parser.add_argument(
"--model_out_dir",
default="teaching_arithmetic/out/addition_plain/",
type=str,
required=False,
help="Path to the model output directory.",
)
args = parser.parse_args()
print("Attempting to save the Space output directory, %s" % args.out_dir)
try:
api.upload_folder(
folder_path=args.out_dir,
path_in_repo=args.out_dir,
repo_id="datacomp/teaching_arithmetic_out_directory",
repo_type="dataset",
)
except Exception as e:
print("That didn't work. Error:")
print(e)
print("Attempting to save the Space model, %s" % args.model_out_dir)
try:
api.upload_folder(
folder_path=args.model_out_dir,
path_in_repo=args.model_out_dir,
repo_id="datacomp/addition_plain",
repo_type="model",
)
except Exception as e:
print("That didn't work. Error:")
print(e) |