Update app.py
Browse files
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
|
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)
|