mohammadreza pakzadian commited on
Commit
cb21c73
·
1 Parent(s): 298ce6b

update model

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -2,7 +2,7 @@ import os
2
  import gradio as gr
3
  import requests
4
  import pandas as pd
5
- from smolagents import CodeAgent, OpenAIServerModel
6
  from tools.final_answer import FinalAnswerTool
7
  from tools.visit_webpage import VisitWebpageTool
8
  from tools.web_search import DuckDuckGoSearchTool
@@ -19,7 +19,12 @@ class BasicAgent:
19
  final_answer = FinalAnswerTool()
20
  visit_webpage = VisitWebpageTool()
21
  web_search = DuckDuckGoSearchTool()
22
- model = OpenAIServerModel(model_id="gpt-4.1", api_key=os.getenv("OPENAI_API_KEY"))
 
 
 
 
 
23
  self.agent = CodeAgent(
24
  model=model,
25
  tools=[visit_webpage, web_search, final_answer],
 
2
  import gradio as gr
3
  import requests
4
  import pandas as pd
5
+ from smolagents import CodeAgent, OpenAIServerModel, HfApiModel
6
  from tools.final_answer import FinalAnswerTool
7
  from tools.visit_webpage import VisitWebpageTool
8
  from tools.web_search import DuckDuckGoSearchTool
 
19
  final_answer = FinalAnswerTool()
20
  visit_webpage = VisitWebpageTool()
21
  web_search = DuckDuckGoSearchTool()
22
+ # model = OpenAIServerModel(model_id="gpt-4.1", api_key=os.getenv("OPENAI_API_KEY"))
23
+ model = HfApiModel(
24
+ max_tokens=2096,
25
+ temperature=0.5,
26
+ model_id='Qwen/Qwen2.5-Coder-7B-Instruct', custom_role_conversions=None,
27
+ )
28
  self.agent = CodeAgent(
29
  model=model,
30
  tools=[visit_webpage, web_search, final_answer],