jesusgj commited on
Commit
0f2240e
·
1 Parent(s): fbd2cf1

Modified files

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -15,7 +15,7 @@ def initialize_agent():
15
 
16
  # 1. Load the model
17
  # Make sure to set HF_TOKEN in your Hugging Face Space secrets
18
- model_name = "HuggingFaceH4/zephyr-7b-beta"
19
  try:
20
  model = InferenceClientModel(model_id=model_name, token=os.environ.get("HF_TOKEN"))
21
  except Exception as e:
@@ -66,7 +66,7 @@ def initialize_agent():
66
  model=model,
67
  managed_agents=[web_agent],
68
  additional_authorized_imports=["time", "numpy", "pandas"],
69
- system_prompt="""You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string."""
70
  )
71
  return manager_agent
72
  else:
 
15
 
16
  # 1. Load the model
17
  # Make sure to set HF_TOKEN in your Hugging Face Space secrets
18
+ model_name = "meta-llama/Llama-3.3-70B-Instruct"
19
  try:
20
  model = InferenceClientModel(model_id=model_name, token=os.environ.get("HF_TOKEN"))
21
  except Exception as e:
 
66
  model=model,
67
  managed_agents=[web_agent],
68
  additional_authorized_imports=["time", "numpy", "pandas"],
69
+ instructions="""You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string."""
70
  )
71
  return manager_agent
72
  else: