Toumaima commited on
Commit
9b378a7
·
verified ·
1 Parent(s): 36e4746

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -20,7 +20,11 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
20
  class BasicAgent:
21
  def __init__(self):
22
  print("BasicAgent initialized.")
23
- self.agent = CodeAgent(tools=[DuckDuckGoSearchTool()], model=HfApiModel())
 
 
 
 
24
 
25
  SYSTEM_PROMPT = """You are a general AI assistant. I will ask you a question. Report your thoughts, and
26
  finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER].
 
20
  class BasicAgent:
21
  def __init__(self):
22
  print("BasicAgent initialized.")
23
+ self.agent = CodeAgent(tools=[DuckDuckGoSearchTool()],
24
+ model=HfApiModel(
25
+ api_key=os.getenv("HF_TOKEN"),
26
+ model_name="mistralai/Mistral-7B-Instruct-v0.1" # Or another accessible model
27
+ ))
28
 
29
  SYSTEM_PROMPT = """You are a general AI assistant. I will ask you a question. Report your thoughts, and
30
  finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER].