A-Aryam commited on
Commit
00f7a0b
·
verified ·
1 Parent(s): d03367b

fix `InferenceClientModel` import err

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -76,7 +76,7 @@ import pytz
76
  import requests
77
  import yaml
78
  import spaces
79
- from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool, FinalAnswerTool, InferenceClientModel
80
  #from tools.final_answer import FinalAnswerTool
81
  from Gradio_UI import GradioUI
82
 
@@ -88,8 +88,8 @@ final_answer = FinalAnswerTool() #final answer tool
88
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
89
 
90
  # our llm:
91
- #model = HfApiModel(
92
- model = InferenceClientModel(
93
  max_tokens=2096,
94
  temperature=0.5,
95
  model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
 
76
  import requests
77
  import yaml
78
  import spaces
79
+ from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool, FinalAnswerTool #, InferenceClientModel
80
  #from tools.final_answer import FinalAnswerTool
81
  from Gradio_UI import GradioUI
82
 
 
88
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
89
 
90
  # our llm:
91
+ model = HfApiModel(
92
+ #model = InferenceClientModel(
93
  max_tokens=2096,
94
  temperature=0.5,
95
  model_id='Qwen/Qwen2.5-Coder-32B-Instruct',