Update tools/final_answer.py

tools/final_answer.py

@@ -1,8 +1,6 @@
 from typing import Any, Optional
 from smolagents.tools import Tool
 
-from smolagents.agent_types import AgentImage
-
 class FinalAnswerTool(Tool):
     name = "final_answer"
     description = "Provides a final answer to the given problem."
@@ -10,9 +8,6 @@ class FinalAnswerTool(Tool):
     output_type = "any"
 
     def forward(self, answer: Any) -> Any:
-        # unwrap AgentImage to raw PIL image
-        if isinstance(answer, AgentImage):
-            return answer.image
         return answer
 
     def __init__(self, *args, **kwargs):