Fix the fallback logic for getting the `SPACE_AUTHOR_NAME` when `user_hf_token` is not set
Browse files
app.py
CHANGED
@@ -1,5 +1,6 @@
|
|
1 |
import logging
|
2 |
import subprocess
|
|
|
3 |
import sys
|
4 |
from dataclasses import dataclass
|
5 |
from pathlib import Path
|
@@ -30,12 +31,13 @@ class Config:
|
|
30 |
def from_env(cls) -> "Config":
|
31 |
"""Create config from environment variables and secrets."""
|
32 |
system_token = st.secrets.get("HF_TOKEN")
|
33 |
-
user_token = st.session_state.get("user_hf_token"
|
34 |
-
|
35 |
-
whoami(token=user_token)["name"]
|
36 |
-
|
37 |
-
|
38 |
-
|
|
|
39 |
hf_token = user_token or system_token
|
40 |
|
41 |
if not hf_token:
|
|
|
1 |
import logging
|
2 |
import subprocess
|
3 |
+
import os
|
4 |
import sys
|
5 |
from dataclasses import dataclass
|
6 |
from pathlib import Path
|
|
|
31 |
def from_env(cls) -> "Config":
|
32 |
"""Create config from environment variables and secrets."""
|
33 |
system_token = st.secrets.get("HF_TOKEN")
|
34 |
+
user_token = st.session_state.get("user_hf_token")
|
35 |
+
if user_token:
|
36 |
+
hf_username = whoami(token=user_token)["name"]
|
37 |
+
else:
|
38 |
+
hf_username = (
|
39 |
+
os.getenv("SPACE_AUTHOR_NAME") or whoami(token=system_token)["name"]
|
40 |
+
)
|
41 |
hf_token = user_token or system_token
|
42 |
|
43 |
if not hf_token:
|