Update app.py

app.py

@@ -4,7 +4,7 @@ import tempfile
 import gradio as gr
 from smolagents import CodeAgent, InferenceClientModel
 from smolagents import HfApiModel
-from transformers import load_tool
+#from transformers import load_tool
 
 #%% Methods
 # Function to add a label to an image
@@ -104,8 +104,8 @@ def generate_object_history(object_name):
 
 #%% Initialization of tools and AI_Agent
 # Import text-to-image tool from Hub
-from smolagents.tools import Tool
-image_generation_tool = load_tool("m-ric/text-to-image", cache=False)
+from smolagents.tools import Tool, load_tool
+image_generation_tool = load_tool("m-ric/text-to-image", cache=False, trust_remote_code=True)
 
 class WrappedTextToImageTool(Tool):
     def run(self, *args, **kwargs):