AMfeta99 commited on
Commit
d91db0a
·
verified ·
1 Parent(s): 0d3c634

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -3,7 +3,7 @@ from PIL import Image, ImageDraw, ImageFont
3
  import tempfile
4
  import gradio as gr
5
  from smolagents import CodeAgent, InferenceClientModel
6
- from smolagents import HfApiEngine
7
  from transformers import load_tool
8
 
9
  #%% Methods
@@ -111,7 +111,8 @@ from transformers.agents.search import DuckDuckGoSearchTool
111
  search_tool = DuckDuckGoSearchTool()
112
 
113
  # Load the LLM engine
114
- llm_engine = HfApiEngine("Qwen/Qwen2.5-72B-Instruct")
 
115
 
116
  # Initialize the agent with both tools
117
  #agent = ReactCodeAgent(tools=[image_generation_tool, search_tool], llm_engine=llm_engine)
 
3
  import tempfile
4
  import gradio as gr
5
  from smolagents import CodeAgent, InferenceClientModel
6
+ from smolagents import HfApiModel
7
  from transformers import load_tool
8
 
9
  #%% Methods
 
111
  search_tool = DuckDuckGoSearchTool()
112
 
113
  # Load the LLM engine
114
+ #llm_engine = HfApiEngine("Qwen/Qwen2.5-72B-Instruct")
115
+ llm_engine =HfApiModel("Qwen/Qwen2.5-72B-Instruct")
116
 
117
  # Initialize the agent with both tools
118
  #agent = ReactCodeAgent(tools=[image_generation_tool, search_tool], llm_engine=llm_engine)