We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent bf8f8ba commit b3c2967Copy full SHA for b3c2967
1 file changed
src/classifier/huggingface/hf_upload_model.py
@@ -4,16 +4,14 @@
4
5
load_dotenv()
6
7
+
8
def upload_model_to_huggingface():
9
HF_TOKEN = os.environ.get("HUGGINGFACE_TOKEN")
10
HF_ENDPOINT = os.environ.get("HUGGINGFACE_ENDPOINT")
11
REPO_ID = os.environ.get("HUGGINGFACE_REPO")
12
MODEL_PATH = os.environ.get("HUGGINGFACE_MODEL_PATH")
13
- api = HfApi(
14
- endpoint=HF_ENDPOINT
15
- token=HF_TOKEN
16
- )
+ api = HfApi(endpoint=HF_ENDPOINT, token=HF_TOKEN)
17
api.upload_file(
18
path_or_fileobj="classifier/models/supervised_hetero.pt",
19
path_in_repo=MODEL_PATH,
0 commit comments