Spaces:
Sleeping
Sleeping
improve: simplify tocken use
Browse files- modules/ai_model.py +1 -0
modules/ai_model.py
CHANGED
|
@@ -40,6 +40,7 @@ class AIModel:
|
|
| 40 |
cache_dir = "/app/.cache/huggingface"
|
| 41 |
login(token=token_to_use, add_to_git_credential=False)
|
| 42 |
log.info("✅ HuggingFace 认证成功")
|
|
|
|
| 43 |
|
| 44 |
|
| 45 |
|
|
|
|
| 40 |
cache_dir = "/app/.cache/huggingface"
|
| 41 |
login(token=token_to_use, add_to_git_credential=False)
|
| 42 |
log.info("✅ HuggingFace 认证成功")
|
| 43 |
+
return token_to_use
|
| 44 |
|
| 45 |
|
| 46 |
|