dianamclean commited on
Commit
8aa6519
·
verified ·
1 Parent(s): 9a692d3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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, get_joke_tool, image_generation_tool], ## add your tools here (don't remove final answer)
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,