Update app.py

app.py

@@ -6,14 +6,14 @@ import pandas as pd
 from smolagents import CodeAgent, DuckDuckGoSearchTool
 from smolagents.models import OpenAIServerModel
 
-#from tools import WikipediaTool, WikipediaSearchTool
-#from tools import WikipediaToolWrapper, WikipediaSearchToolWrapper
-
 from smolagents import Tool
 from wikipedia_searcher import WikipediaSearcher
 
 from audio_transcriber import AudioTranscriptionTool
 
+from image_analyzer import ImageAnalyzer
+image_tool = ImageAnalyzer()
+
 
 class WikipediaSearchTool(Tool):
     name = "wikipedia_search"
@@ -58,11 +58,12 @@ class PatchedOpenAIServerModel(OpenAIServerModel):
 
 class MyAgent:
     def __init__(self):
-        self.model = PatchedOpenAIServerModel(model_id="gpt-4-turbo")
+        self.model = PatchedOpenAIServerModel(model_id="gpt-4-vision-preview") #gpt-4-turbo
         self.agent = CodeAgent(tools=[
             DuckDuckGoSearchTool(),
             wikipedia_search_tool,
-            AudioTranscriptionTool()
+            AudioTranscriptionTool(),
+            ImageAnalyzer()
         ], model=self.model)
 
     def __call__(self, task: dict) -> str:
@@ -211,3 +212,4 @@ if __name__ == "__main__":
     print("-"*(60 + len(" App Starting ")) + "\n")
     print("Launching Gradio Interface for Basic Agent Evaluation...")
     demo.launch(debug=True, share=False)
+