thatting commited on
Commit
91223dd
·
verified ·
1 Parent(s): 3e73630

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -12,9 +12,9 @@ OPENAI_KEY = os.getenv("OPEN_AI_KEY")
12
 
13
  class Agent:
14
  def __init__(self):
15
- self.hf_token = os.getenv("HF_TOKEN")
16
 
17
- self.model = InferenceClientModel(
18
  model_id="Qwen/Qwen3-235B-A22B",
19
  token=HF_TOKEN
20
  )
 
12
 
13
  class Agent:
14
  def __init__(self):
15
+ self.hf_token = os.getenv("HF_API_KEY")
16
 
17
+ self.model = HfApiModel(
18
  model_id="Qwen/Qwen3-235B-A22B",
19
  token=HF_TOKEN
20
  )