Update tools/final_answer.py

tools/final_answer.py

@@ -1,5 +1,9 @@
+
+
+
+
 from typing import Any
-from smolagents.tools import Tool, AgentImage
+from smolagents.tools import Tool
 
 class FinalAnswerTool(Tool):
     name = "final_answer"
@@ -12,7 +16,7 @@ class FinalAnswerTool(Tool):
 
     def forward(self, answer: Any = "", image: Any = None) -> str:
         # Handle AgentImage rendering
-        if isinstance(image, AgentImage):
+        if isinstance(image):
             try:
                 image_html = image.display()  # or image.to_html()
             except: