Aharneish commited on
Commit
05d59f0
·
verified ·
1 Parent(s): 9ab745c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
11
  # --- Define Agent ---
12
  class SmolAgentWrapper:
13
  def __init__(self):
14
- self.model = TransformersModel(model_id="gpt2") # Uses flan-t5-base via HF Inference API
15
  self.tools = [DuckDuckGoSearchTool()] # You can add more tools here
16
  self.agent = CodeAgent(model=self.model, tools=self.tools)
17
 
 
11
  # --- Define Agent ---
12
  class SmolAgentWrapper:
13
  def __init__(self):
14
+ self.model = TransformersModel(model_id="facebook/blenderbot-400M-distill")
15
  self.tools = [DuckDuckGoSearchTool()] # You can add more tools here
16
  self.agent = CodeAgent(model=self.model, tools=self.tools)
17