Update app.py (#5)

- Update app.py (33047badac82eff7667491ba66f3477c443888be)

app.py

@@ -6,8 +6,8 @@ import pandas as pd
 
 # Import our custom tools from their modules
 from huggingface_hub import login
-from smolagents import CodeAgent, InferenceClientModel, OpenAIServerModel, GoogleSearchTool
-from tools import web_search, visit_webpage, final_answer
+from smolagents import CodeAgent, InferenceClientModel, OpenAIServerModel
+from tools import web_search, visit_webpage, final_answer, image_qa
 # from tools import web_search, visit_webpage, final_answer, go_back, close_popups, search_item_ctrl_f
 from retriever import load_guest_dataset
 from functools import lru_cache
@@ -29,12 +29,14 @@ class BasicAgent:
         login(token=api_key)
         
         # Create the search tool
-        # self.web_search = web_search
+        self.web_search = web_search
 
         self.visit_webpage = visit_webpage
 
         self.final_answer = final_answer
 
+        self.image_qa = image_qa
+
         # self.go_back = go_back 
         
         # self.close_popups = close_popups
@@ -42,8 +44,8 @@ class BasicAgent:
         # self.search_item_ctrl_f = search_item_ctrl_f
         
         # Create the model
-        # self.model = InferenceClientModel(model_id="Qwen/Qwen2.5-72B-Instruct")
-        self.model = InferenceClientModel(model_id="Qwen/Qwen2.5-7B-Instruct") #smaller/faster
+        self.model = InferenceClientModel(model_id="Qwen/Qwen2.5-72B-Instruct")
+        # self.model = InferenceClientModel(model_id="Qwen/Qwen2.5-7B-Instruct") #smaller/faster
         # self.model = OpenAIServerModel(model_id="gpt-4o")
 
         # Define your system prompt
@@ -56,7 +58,7 @@ class BasicAgent:
         
         # Create the agent with tools
         self.agent = CodeAgent(
-            tools=[GoogleSearchTool(), self.visit_webpage, self.final_answer],
+            tools=[web_search, self.visit_webpage, self.final_answer, self.image_qa],
             # tools=[self.web_search, self.visit_webpage, self.final_answer, go_back, close_popups, search_item_ctrl_f],
             model=self.model,
             additional_authorized_imports=["pandas", "openpyxl", "yt_dlp", "requests", "io", "json", "whisper", "bs4"],