Update tools/final_answer.py

tools/final_answer.py

@@ -1,5 +1,6 @@
 from typing import Any, Optional
 from smolagents.tools import Tool
+from PIL import Image
 
 class FinalAnswerTool(Tool):
     name = "final_answer"
@@ -8,6 +9,8 @@ class FinalAnswerTool(Tool):
     output_type = "any"
 
     def forward(self, answer: Any) -> Any:
+        if isinstance(answer, Image.Image):
+            answer.show()
         return answer
 
     def __init__(self, *args, **kwargs):