Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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=
|
| 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 |
|