Update app.py

app.py

@@ -4,9 +4,6 @@ import streamlit as st
 from smolagents import load_tool, CodeAgent, HfApiModel, DuckDuckGoSearchTool
 from huggingface_hub import InferenceClient
 
-# Import the image generation tool from the Hub
-image_generation_tool = load_tool("m-ric/text-to-image", trust_remote_code=True)
-
 # Instantiate the DuckDuckGo search tool from SmolAgents
 search_tool = DuckDuckGoSearchTool()
 
@@ -21,8 +18,8 @@ client = InferenceClient(token=hf_token)
 # Initialize the SmolAgent model
 model = HfApiModel(model_id="meta-llama/Llama-3.2-3B-Instruct", token=hf_token)
 
-# Create the agent with both the search and image generation tools
-agent = CodeAgent(tools=[search_tool, image_generation_tool], model=model)
+# Create the agent with only the search tool
+agent = CodeAgent(tools=[search_tool], model=model)
 
 def apply_custom_styles():
     st.markdown(
@@ -93,11 +90,11 @@ def main():
         user_input = st.session_state["user_input"]
         if user_input.strip():
             try:
-                # Instruct Tom Riddle to use web search efficiently only when unsure.
+                # Instruct Tom Riddle to use the search tool only when uncertain, efficiently.
                 response = agent.run(
                     f"You are Tom Riddle, a cunning and enigmatic character from Harry Potter. "
                     f"Answer the user's query clearly and concisely in your distinct persona. "
-                    f"If you are not certain of your answer, only then use your available tools (DuckDuckGo search and image generation) to quickly gather necessary insights—doing so efficiently without causing delays. "
+                    f"If you are not certain of your answer, only then use your available DuckDuckGo search tool to quickly gather the necessary insights—doing so efficiently without causing delays. "
                     f"Finally, produce your answer in your characteristic voice without revealing your internal process. "
                     f"User Query: {user_input}"
                 )