Spaces:
Sleeping
Sleeping
feat: hf token
Browse files
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
|
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,
|