mrp-78 commited on
Commit
589a46b
·
verified ·
1 Parent(s): 219207a

Update app.py

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, HfApiModel, load_tool, tool
7
  from tools.final_answer import FinalAnswerTool
8
  from tools.visit_webpage import VisitWebpageTool
9
  from tools.web_search import DuckDuckGoSearchTool
@@ -21,7 +21,7 @@ class BasicAgent:
21
  final_answer = FinalAnswerTool()
22
  visit_webpage = VisitWebpageTool()
23
  web_search = DuckDuckGoSearchTool()
24
- model = HfApiModel(
25
  max_tokens=2096,
26
  temperature=0.5,
27
  model_id='Qwen/Qwen2.5-Coder-32B-Instruct', custom_role_conversions=None,
 
3
  import requests
4
  import inspect
5
  import pandas as pd
6
+ from smolagents import CodeAgent, InferenceClientModel, load_tool, tool
7
  from tools.final_answer import FinalAnswerTool
8
  from tools.visit_webpage import VisitWebpageTool
9
  from tools.web_search import DuckDuckGoSearchTool
 
21
  final_answer = FinalAnswerTool()
22
  visit_webpage = VisitWebpageTool()
23
  web_search = DuckDuckGoSearchTool()
24
+ model = InferenceClientModel(
25
  max_tokens=2096,
26
  temperature=0.5,
27
  model_id='Qwen/Qwen2.5-Coder-32B-Instruct', custom_role_conversions=None,