Update app.py
Browse files
app.py
CHANGED
@@ -105,6 +105,13 @@ def generate_object_history(object_name):
|
|
105 |
#%% Initialization of tools and AI_Agent
|
106 |
# Import text-to-image tool from Hub
|
107 |
from smolagents.tools import Tool, load_tool
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
108 |
image_generation_tool = load_tool("m-ric/text-to-image", cache=False, trust_remote_code=True)
|
109 |
|
110 |
|
@@ -117,7 +124,7 @@ class WrappedTextToImageTool(Tool):
|
|
117 |
return image_generation_tool(prompt)
|
118 |
|
119 |
image_generation_tool = WrappedTextToImageTool()
|
120 |
-
|
121 |
# Import search tool from LangChain
|
122 |
from transformers.agents.search import DuckDuckGoSearchTool
|
123 |
search_tool = DuckDuckGoSearchTool()
|
|
|
105 |
#%% Initialization of tools and AI_Agent
|
106 |
# Import text-to-image tool from Hub
|
107 |
from smolagents.tools import Tool, load_tool
|
108 |
+
image_generation_tool = Tool.from_space(
|
109 |
+
space_id="m-ric/text-to-image",
|
110 |
+
name="text-to-image-generator",
|
111 |
+
description="Generate an image from a prompt"
|
112 |
+
)
|
113 |
+
'''
|
114 |
+
|
115 |
image_generation_tool = load_tool("m-ric/text-to-image", cache=False, trust_remote_code=True)
|
116 |
|
117 |
|
|
|
124 |
return image_generation_tool(prompt)
|
125 |
|
126 |
image_generation_tool = WrappedTextToImageTool()
|
127 |
+
'''
|
128 |
# Import search tool from LangChain
|
129 |
from transformers.agents.search import DuckDuckGoSearchTool
|
130 |
search_tool = DuckDuckGoSearchTool()
|