chankhavu commited on
Commit
eccb556
·
verified ·
1 Parent(s): a0e459e

daemon: support gradio_client token kwarg rename

Browse files
Files changed (1) hide show
  1. 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 = Client(SPACE, hf_token=HF_TOKEN, verbose=False)
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