wsj1995 commited on
Commit
ef0cccb
·
1 Parent(s): 46c5d91

feat: hf token

Browse files
Files changed (1) hide show
  1. api.py +2 -3
api.py CHANGED
@@ -8,7 +8,7 @@ from io import BytesIO
8
  from datetime import datetime
9
  import hashlib
10
  import time
11
- from huggingface_hub import HfApi, HfFolder
12
 
13
  API_AUTH_KEY = os.environ.get('AUTH_KEY')
14
 
@@ -67,8 +67,7 @@ async def upload_file(userId: str, modelId: str, filename: str, file: UploadFile
67
  buffer.write(await file.read())
68
  callback(modelId,'UPLOADING')
69
  pathInRepo = f"{userId}/{filename}"
70
- huggingfaceApi = HfApi()
71
- HfFolder.save_token(HUGGINGFACE_API_TOKEN)
72
  huggingfaceApi.upload_file(
73
  path_or_fileobj=file_location,
74
  path_in_repo=pathInRepo,
 
8
  from datetime import datetime
9
  import hashlib
10
  import time
11
+ from huggingface_hub import HfApi
12
 
13
  API_AUTH_KEY = os.environ.get('AUTH_KEY')
14
 
 
67
  buffer.write(await file.read())
68
  callback(modelId,'UPLOADING')
69
  pathInRepo = f"{userId}/{filename}"
70
+ huggingfaceApi = HfApi(token=api_token)
 
71
  huggingfaceApi.upload_file(
72
  path_or_fileobj=file_location,
73
  path_in_repo=pathInRepo,