Update image generation tool in app.py to use m-ric/text-to-image for improved functionality

app.py

@@ -54,7 +54,9 @@ model = HfApiModel(
 
 
 # Import tool from Hub
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+# image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+# m-ric/text-to-image
+image_generation_tool = load_tool("m-ric/text-to-image", trust_remote_code=True)
 
 with open("prompts.yaml", "r") as stream:
     prompt_templates = yaml.safe_load(stream)

tools/final_answer.py

@@ -8,12 +8,6 @@ class FinalAnswerTool(Tool):
     output_type = "any"
 
     def forward(self, answer: Any) -> Any:
-        # handle image, e.g. smolagents.agent_types.AgentImage
-        if hasattr(answer, "_raw") and hasattr(answer._raw, "save"):
-            import tempfile
-            with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as f:
-                answer._raw.save(f.name, format="PNG")
-                answer._path = f.name
         return answer
 
     def __init__(self, *args, **kwargs):