SeaWolf-AI commited on
Commit
fd2efdc
·
verified ·
1 Parent(s): ed41730

gradio_client token 양쪽대응

Browse files
Files changed (2) hide show
  1. app.py +5 -1
  2. requirements.txt +0 -1
app.py CHANGED
@@ -17,9 +17,13 @@ CONSULT = os.environ.get("CHITOS_CONSULT_ENDPOINT", "")
17
  _eng_client = None
18
 
19
  def _engine():
 
20
  global _eng_client
21
  if _eng_client is None and ENGINE:
22
- _eng_client = Client(ENGINE, hf_token=HFTOKEN or None)
 
 
 
23
  return _eng_client
24
 
25
  GCOL = {"A+": "#067647", "A": "#067647", "B": "#2563eb", "C": "#b54708", "D": "#b54708", "F": "#b42318"}
 
17
  _eng_client = None
18
 
19
  def _engine():
20
+ global _eng_client
21
  global _eng_client
22
  if _eng_client is None and ENGINE:
23
+ try:
24
+ _eng_client = Client(ENGINE, token=HFTOKEN or None)
25
+ except TypeError:
26
+ _eng_client = Client(ENGINE, hf_token=HFTOKEN or None)
27
  return _eng_client
28
 
29
  GCOL = {"A+": "#067647", "A": "#067647", "B": "#2563eb", "C": "#b54708", "D": "#b54708", "F": "#b42318"}
requirements.txt CHANGED
@@ -1,3 +1,2 @@
1
  gradio==5.9.1
2
  requests==2.32.3
3
- gradio_client==1.5.2
 
1
  gradio==5.9.1
2
  requests==2.32.3