sqfoo commited on
Commit
93bfc18
·
verified ·
1 Parent(s): e5ce92e

replace gpt-4o with other vlm

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -10,10 +10,13 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
10
 
11
  # --- Basic Agent Definition ---
12
  # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
13
- from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel
 
 
14
  class BasicAgent:
15
  def __init__(self):
16
- model = OpenAIServerModel(model_id="gpt-4o")
 
17
  self.agent = CodeAgent(tools=[DuckDuckGoSearchTool()], model=model)
18
  print("BasicAgent initialized.")
19
  def __call__(self, question: str) -> str:
 
10
 
11
  # --- Basic Agent Definition ---
12
  # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
13
+ from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel, VLLMModel
14
+
15
+
16
  class BasicAgent:
17
  def __init__(self):
18
+ # model = OpenAIServerModel(model_id="gpt-4o")
19
+ model = VLLMModel(model_id="HuggingFaceTB/SmolLM-135M-Instruct")
20
  self.agent = CodeAgent(tools=[DuckDuckGoSearchTool()], model=model)
21
  print("BasicAgent initialized.")
22
  def __call__(self, question: str) -> str: