dlaima commited on
Commit
d07e368
·
verified ·
1 Parent(s): b30995e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -58,11 +58,14 @@ tools = [audio_tool, image_tool, wiki_tool]
58
  # )
59
  # super().__init__(model=model, tools=tools, system_prompt=SYSTEM_PROMPT)
60
 
 
 
 
61
  class MyAgent(CodeAgent):
62
  def __init__(self):
63
  model = HfApiModel(
64
  model="mistralai/Mixtral-8x7B-Instruct-v0.1", # pass model ID, not full URL
65
- api_key=os.getenv("HF_API_TOKEN")
66
  )
67
  super().__init__(model=model, tools=tools)
68
 
 
58
  # )
59
  # super().__init__(model=model, tools=tools, system_prompt=SYSTEM_PROMPT)
60
 
61
+ api_key = os.getenv("HF_API_TOKEN", "").strip()
62
+
63
+
64
  class MyAgent(CodeAgent):
65
  def __init__(self):
66
  model = HfApiModel(
67
  model="mistralai/Mixtral-8x7B-Instruct-v0.1", # pass model ID, not full URL
68
+ api_key=api_key
69
  )
70
  super().__init__(model=model, tools=tools)
71