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 HfApiModel
|
7 |
#from transformers import load_tool
|
8 |
|
9 |
#%% Methods
|
@@ -126,7 +126,7 @@ class WrappedTextToImageTool(Tool):
|
|
126 |
image_generation_tool = WrappedTextToImageTool()
|
127 |
'''
|
128 |
# Import search tool from LangChain
|
129 |
-
from transformers.agents.search import DuckDuckGoSearchTool
|
130 |
search_tool = DuckDuckGoSearchTool()
|
131 |
|
132 |
# Load the LLM engine
|
|
|
3 |
import tempfile
|
4 |
import gradio as gr
|
5 |
from smolagents import CodeAgent, InferenceClientModel
|
6 |
+
from smolagents import HfApiModel, DuckDuckGoSearchTool
|
7 |
#from transformers import load_tool
|
8 |
|
9 |
#%% Methods
|
|
|
126 |
image_generation_tool = WrappedTextToImageTool()
|
127 |
'''
|
128 |
# Import search tool from LangChain
|
129 |
+
#from transformers.agents.search import DuckDuckGoSearchTool
|
130 |
search_tool = DuckDuckGoSearchTool()
|
131 |
|
132 |
# Load the LLM engine
|