TommasoBB commited on
Commit
87a780b
·
verified ·
1 Parent(s): 5dc38a3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,7 +12,7 @@ import base64
12
  import urllib.request
13
 
14
  from huggingface_hub import InferenceClient
15
- from smolagents import CodeAgent, HfApiModel, tool
16
  from smolagents import DuckDuckGoSearchTool, VisitWebpageTool
17
 
18
 
@@ -278,7 +278,7 @@ def build_agent() -> CodeAgent:
278
  - read_task_file : download & parse task attachments
279
  (PDF, CSV, Excel, text, image, audio, video)
280
  """
281
- model = HfApiModel(
282
  model_id="Qwen/Qwen2.5-72B-Instruct",
283
  token=os.getenv("HF_API_TOKEN"),
284
  )
 
12
  import urllib.request
13
 
14
  from huggingface_hub import InferenceClient
15
+ from smolagents import CodeAgent, InferenceClientModel, tool
16
  from smolagents import DuckDuckGoSearchTool, VisitWebpageTool
17
 
18
 
 
278
  - read_task_file : download & parse task attachments
279
  (PDF, CSV, Excel, text, image, audio, video)
280
  """
281
+ model = InferenceClientModel(
282
  model_id="Qwen/Qwen2.5-72B-Instruct",
283
  token=os.getenv("HF_API_TOKEN"),
284
  )