rca-umb commited on
Commit
773d28f
·
verified ·
1 Parent(s): 61aecb6

Switch to transformers model

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,7 +3,7 @@ import gradio as gr
3
  import requests
4
  import inspect
5
  import pandas as pd
6
- from smolagents import CodeAgent, DuckDuckGoSearchTool, VisitWebpageTool, InferenceClientModel
7
 
8
  # (Keep Constants as is)
9
  # --- Constants ---
@@ -14,7 +14,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
14
  class MyAgent(CodeAgent):
15
  def __init__(self):
16
  super().__init__(
17
- model=InferenceClientModel(),
18
  tools=[
19
  DuckDuckGoSearchTool(),
20
  VisitWebpageTool()
 
3
  import requests
4
  import inspect
5
  import pandas as pd
6
+ from smolagents import CodeAgent, DuckDuckGoSearchTool, VisitWebpageTool, TransformersModel
7
 
8
  # (Keep Constants as is)
9
  # --- Constants ---
 
14
  class MyAgent(CodeAgent):
15
  def __init__(self):
16
  super().__init__(
17
+ model = TransformersModel(model_id="HuggingFaceTB/SmolLM-135M-Instruct"),
18
  tools=[
19
  DuckDuckGoSearchTool(),
20
  VisitWebpageTool()