Visionkambanje commited on
Commit
dcd737d
·
verified ·
1 Parent(s): 2faef85

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -5,7 +5,7 @@ import inspect
5
  import pandas as pd
6
  import re
7
 
8
- from smolagents import CodeAgent, DuckDuckGoSearchTool,HfApiModel , PythonInterpreterTool, WikipediaSearchTool, FinalAnswerTool
9
 
10
  # (Keep Constants as is)
11
  # --- Constants ---
@@ -22,9 +22,9 @@ agent = CodeAgent(
22
  )
23
 
24
  import os
25
- from smolagents import HfApiModel
26
 
27
- model = HfApiModel(
28
  model_id="Qwen/Qwen1.5-1.8B-Chat", # example HF chat model
29
  token=os.getenv("HF_TOKEN") # only needed if private model
30
  )
 
5
  import pandas as pd
6
  import re
7
 
8
+ from smolagents import CodeAgent, DuckDuckGoSearchTool,InferenceClientModel , PythonInterpreterTool, WikipediaSearchTool, FinalAnswerTool
9
 
10
  # (Keep Constants as is)
11
  # --- Constants ---
 
22
  )
23
 
24
  import os
25
+ from smolagents import InferenceClientModel
26
 
27
+ model = InferenceClientModel(
28
  model_id="Qwen/Qwen1.5-1.8B-Chat", # example HF chat model
29
  token=os.getenv("HF_TOKEN") # only needed if private model
30
  )