lization commited on
Commit
58e69bd
·
verified ·
1 Parent(s): 295ceb0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -45,6 +45,7 @@ def get_current_time_in_timezone(timezone: str) -> str:
45
 
46
  final_answer = FinalAnswerTool()
47
 
 
48
  model = HfApiModel(
49
  max_tokens=2096,
50
  temperature=0.5,
@@ -52,12 +53,14 @@ model = HfApiModel(
52
  custom_role_conversions=None,
53
  )
54
 
 
55
  # Import tool from Hub
56
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
57
 
58
  # Load system prompt from prompt.yaml file
59
  with open("prompts.yaml", 'r') as stream:
60
  prompt_templates = yaml.safe_load(stream)
 
61
 
62
  agent = CodeAgent(
63
  model=model,
 
45
 
46
  final_answer = FinalAnswerTool()
47
 
48
+
49
  model = HfApiModel(
50
  max_tokens=2096,
51
  temperature=0.5,
 
53
  custom_role_conversions=None,
54
  )
55
 
56
+
57
  # Import tool from Hub
58
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
59
 
60
  # Load system prompt from prompt.yaml file
61
  with open("prompts.yaml", 'r') as stream:
62
  prompt_templates = yaml.safe_load(stream)
63
+
64
 
65
  agent = CodeAgent(
66
  model=model,