Thanh Vinh Vo commited on
Commit
80d62d5
·
1 Parent(s): 7caef0c
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -28,9 +28,8 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
28
  class BasicAgent:
29
  def __init__(self):
30
  print("BasicAgent initialized.")
31
- mOpenAI = OpenAIServerModel(model_id="gpt-4o")
32
- mQwen = InferenceClientModel(
33
- model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
34
  )
35
 
36
  self.agent = CodeAgent(
@@ -38,7 +37,7 @@ class BasicAgent:
38
  VisitWebpageTool(),
39
  DuckDuckGoSearchTool(),
40
  ],
41
- model=mOpenAI,
42
  additional_authorized_imports=[
43
  "requests",
44
  "bs4",
 
28
  class BasicAgent:
29
  def __init__(self):
30
  print("BasicAgent initialized.")
31
+ model = InferenceClientModel(
32
+ model_id="meta-llama/Llama-3.2-90B-Vision-Instruct",
 
33
  )
34
 
35
  self.agent = CodeAgent(
 
37
  VisitWebpageTool(),
38
  DuckDuckGoSearchTool(),
39
  ],
40
+ model=model,
41
  additional_authorized_imports=[
42
  "requests",
43
  "bs4",