Removed query image tool.

app.py

@@ -8,7 +8,7 @@ from tools.exchange_rates import ExchangeRatesTool
 from tools.web_search import ddgs
 from tools.visit_webpage import VisitWebpageTool
 from Gradio_UI import GradioUI
-from tools.new_tools import QueryImageTool, ReverseStringTool, WikiSearchTool
+from tools.new_tools import ReverseStringTool, WikiSearchTool
 from tools.get_current_time import GetCurrentTimeTool
 
 final_answer = FinalAnswerTool()
@@ -24,15 +24,12 @@ custom_role_conversions=None,
 )
 
 
-# Import tool from Hub
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, ddgs(), VisitWebpageTool(), ExchangeRatesTool(), WikiSearchTool, ReverseStringTool(), GetCurrentTimeTool()], ## add your tools here (don't remove final answer)
+    tools=[final_answer, ddgs(), VisitWebpageTool(), ExchangeRatesTool(), WikiSearchTool(), ReverseStringTool(), GetCurrentTimeTool()], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,

tools/new_tools.py

@@ -2,17 +2,7 @@ from smolagents import Tool
 from transformers import AutoProcessor
   # adjust import to your setup
 # from langchain_community.retrievers import WikipediaRetriever  # or your existing retriever
-try:
-    import torch
-    from transformers import AutoProcessor, Qwen2_5_VLForConditionalGeneration
-    from qwen_vl_utils import process_vision_info
-    HAS_VISION_DEPS = True
-except Exception:
-    torch = None
-    AutoProcessor = None
-    Qwen2_5_VLForConditionalGeneration = None
-    process_vision_info = None
-    HAS_VISION_DEPS = False
+
 
 class WikiSearchTool(Tool):
     name = "wiki_search"