IamRulo commited on
Commit
e3d3777
·
verified ·
1 Parent(s): 0f2c4b4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -11,7 +11,6 @@ from smolagents import(
11
  DuckDuckGoSearchTool,
12
  InferenceClientModel,
13
  HfApiModel,
14
- OpenAIServerModel
15
  )
16
  # Rulo
17
 
@@ -24,10 +23,11 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
24
  class BasicAgent:
25
  def __init__(self):
26
  #rulo copycat
27
- self.model = OpenAIServerModel(
28
- model_id='o4-mini',
29
- #api_base="https://api.openai.com/v1",
30
- api_key=os.environ["OPENAI_API_KEY"],
 
31
  )
32
  self.agent = ToolCallingAgent(
33
  tools=[DuckDuckGoSearchTool()],
 
11
  DuckDuckGoSearchTool,
12
  InferenceClientModel,
13
  HfApiModel,
 
14
  )
15
  # Rulo
16
 
 
23
  class BasicAgent:
24
  def __init__(self):
25
  #rulo copycat
26
+ self.model = HfApiModel(
27
+ max_tokens=150,
28
+ temperature=0.3,
29
+ model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
30
+ custom_role_conversion=None,
31
  )
32
  self.agent = ToolCallingAgent(
33
  tools=[DuckDuckGoSearchTool()],