antonyjames commited on
Commit
36f94c7
·
verified ·
1 Parent(s): 664f2aa

Upload agent

Browse files
Files changed (2) hide show
  1. agent.json +4 -5
  2. app.py +3 -4
agent.json CHANGED
@@ -3,12 +3,11 @@
3
  "final_answer"
4
  ],
5
  "model": {
6
- "class": "HfApiModel",
7
  "data": {
8
- "last_input_token_count": 4852,
9
- "last_output_token_count": 417,
10
- "model_id": "Qwen/Qwen2.5-Coder-32B-Instruct",
11
- "provider": null
12
  }
13
  },
14
  "managed_agents": {},
 
3
  "final_answer"
4
  ],
5
  "model": {
6
+ "class": "OpenAIServerModel",
7
  "data": {
8
+ "last_input_token_count": 2416,
9
+ "last_output_token_count": 69,
10
+ "model_id": "gpt-4o-mini"
 
11
  }
12
  },
13
  "managed_agents": {},
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import yaml
2
  import os
3
- from smolagents import GradioUI, CodeAgent, HfApiModel
4
 
5
  # Get current directory path
6
  CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
@@ -9,9 +9,8 @@ from tools.final_answer import FinalAnswerTool as FinalAnswer
9
 
10
 
11
 
12
- model = HfApiModel(
13
- model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
14
- provider=None,
15
  )
16
 
17
  final_answer = FinalAnswer()
 
1
  import yaml
2
  import os
3
+ from smolagents import GradioUI, CodeAgent, OpenAIServerModel
4
 
5
  # Get current directory path
6
  CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
 
9
 
10
 
11
 
12
+ model = OpenAIServerModel(
13
+ model_id='gpt-4o-mini',
 
14
  )
15
 
16
  final_answer = FinalAnswer()