Geevarghese George commited on
Commit
c8ed8e6
·
1 Parent(s): 8f48d14

fix token obtain

Browse files
Files changed (1) hide show
  1. app.py +4 -6
app.py CHANGED
@@ -43,13 +43,11 @@ _monkeypatch_gradio_save_history()
43
 
44
 
45
  def get_agent_model(model_name: str, oauth_token: gr.OAuthToken = None):
46
- token = os.getenv("HF_TOKEN", None) or oauth_token.token if oauth_token else None
47
  # provider = os.getenv("HF_INFERENCE_PROVIDER", "together")
48
- model = InferenceClientModel(
49
- # token=token,
50
- model_id=model_name,
51
- # provider=provider,
52
- api_key=token,
53
  )
54
  return model
55
 
 
43
 
44
 
45
  def get_agent_model(model_name: str, oauth_token: gr.OAuthToken = None):
46
+ token = oauth_token.token if oauth_token else None
47
  # provider = os.getenv("HF_INFERENCE_PROVIDER", "together")
48
+ model = InferenceClientModel(
49
+ model_id=model_name,
50
+ token=token,
 
 
51
  )
52
  return model
53