Skip to content

Commit b3c2967

Browse files
committed
fix: typo
1 parent bf8f8ba commit b3c2967

1 file changed

Lines changed: 2 additions & 4 deletions

File tree

src/classifier/huggingface/hf_upload_model.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,16 +4,14 @@
44

55
load_dotenv()
66

7+
78
def upload_model_to_huggingface():
89
HF_TOKEN = os.environ.get("HUGGINGFACE_TOKEN")
910
HF_ENDPOINT = os.environ.get("HUGGINGFACE_ENDPOINT")
1011
REPO_ID = os.environ.get("HUGGINGFACE_REPO")
1112
MODEL_PATH = os.environ.get("HUGGINGFACE_MODEL_PATH")
1213

13-
api = HfApi(
14-
endpoint=HF_ENDPOINT
15-
token=HF_TOKEN
16-
)
14+
api = HfApi(endpoint=HF_ENDPOINT, token=HF_TOKEN)
1715
api.upload_file(
1816
path_or_fileobj="classifier/models/supervised_hetero.pt",
1917
path_in_repo=MODEL_PATH,

0 commit comments

Comments
 (0)