Spaces:
Running
Running
daemon: support gradio_client token kwarg rename
Browse files- daemon/client.py +9 -1
daemon/client.py
CHANGED
|
@@ -155,11 +155,19 @@ def get_shell(name: str) -> Shell:
|
|
| 155 |
# --------------------------------------------------------------------------- #
|
| 156 |
# Relay client #
|
| 157 |
# --------------------------------------------------------------------------- #
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 158 |
def connect() -> Client:
|
| 159 |
while True:
|
| 160 |
try:
|
| 161 |
log(f"connecting to Space {SPACE} …")
|
| 162 |
-
client =
|
| 163 |
client.predict(CLIENT_ID, SECRET, _meta(), api_name="/register")
|
| 164 |
log(f"registered as '{CLIENT_ID}'")
|
| 165 |
return client
|
|
|
|
| 155 |
# --------------------------------------------------------------------------- #
|
| 156 |
# Relay client #
|
| 157 |
# --------------------------------------------------------------------------- #
|
| 158 |
+
def _make_client() -> Client:
|
| 159 |
+
# gradio_client renamed the auth kwarg from `hf_token` to `token` (~v2.x).
|
| 160 |
+
try:
|
| 161 |
+
return Client(SPACE, token=HF_TOKEN, verbose=False)
|
| 162 |
+
except TypeError:
|
| 163 |
+
return Client(SPACE, hf_token=HF_TOKEN, verbose=False)
|
| 164 |
+
|
| 165 |
+
|
| 166 |
def connect() -> Client:
|
| 167 |
while True:
|
| 168 |
try:
|
| 169 |
log(f"connecting to Space {SPACE} …")
|
| 170 |
+
client = _make_client()
|
| 171 |
client.predict(CLIENT_ID, SECRET, _meta(), api_name="/register")
|
| 172 |
log(f"registered as '{CLIENT_ID}'")
|
| 173 |
return client
|