Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -52,6 +52,7 @@ def get_joke_tool(arg1:str)-> str: #it's import to specify the return type
|
|
| 52 |
|
| 53 |
|
| 54 |
final_answer = FinalAnswerTool()
|
|
|
|
| 55 |
model = HfApiModel(
|
| 56 |
max_tokens=2096,
|
| 57 |
temperature=0.5,
|
|
@@ -68,7 +69,7 @@ with open("prompts.yaml", 'r') as stream:
|
|
| 68 |
|
| 69 |
agent = CodeAgent(
|
| 70 |
model=model,
|
| 71 |
-
tools=[final_answer, get_current_time_in_timezone,
|
| 72 |
print tools, ## new
|
| 73 |
max_steps=6,
|
| 74 |
verbosity_level=1,
|
|
|
|
| 52 |
|
| 53 |
|
| 54 |
final_answer = FinalAnswerTool()
|
| 55 |
+
|
| 56 |
model = HfApiModel(
|
| 57 |
max_tokens=2096,
|
| 58 |
temperature=0.5,
|
|
|
|
| 69 |
|
| 70 |
agent = CodeAgent(
|
| 71 |
model=model,
|
| 72 |
+
tools=[final_answer, get_current_time_in_timezone, image_generation_tool], ## add your tools here (don't remove final answer)
|
| 73 |
print tools, ## new
|
| 74 |
max_steps=6,
|
| 75 |
verbosity_level=1,
|