AMfeta99 commited on
Commit
b4adb3f
·
verified ·
1 Parent(s): 2345fb7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
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()