susmitsil commited on
Commit
e78be38
·
verified ·
1 Parent(s): 4bee79c
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -81,7 +81,8 @@ class BasicAgent:
81
  if hf_token:
82
  self.gaia_agent = GAIAAgent(
83
  model_type="HfApiModel",
84
- model_id="mistralai/Mistral-7B-Instruct-v0.2",
 
85
  api_key=hf_token,
86
  temperature=temperature,
87
  executor_type="local",
@@ -103,7 +104,8 @@ class BasicAgent:
103
  print(f"OpenAI module error: {ie}. Falling back to HfApiModel.")
104
  self.gaia_agent = GAIAAgent(
105
  model_type="HfApiModel",
106
- model_id="mistralai/Mistral-7B-Instruct-v0.2",
 
107
  api_key=hf_token,
108
  temperature=temperature,
109
  executor_type="local",
 
81
  if hf_token:
82
  self.gaia_agent = GAIAAgent(
83
  model_type="HfApiModel",
84
+ # model_id="mistralai/Mistral-7B-Instruct-v0.2",
85
+ model_id="mistralai/Mistral-7B-Instruct-v0.1",
86
  api_key=hf_token,
87
  temperature=temperature,
88
  executor_type="local",
 
104
  print(f"OpenAI module error: {ie}. Falling back to HfApiModel.")
105
  self.gaia_agent = GAIAAgent(
106
  model_type="HfApiModel",
107
+ # model_id="mistralai/Mistral-7B-Instruct-v0.2",
108
+ model_id="mistralai/Mistral-7B-Instruct-v0.1",
109
  api_key=hf_token,
110
  temperature=temperature,
111
  executor_type="local",