Update app.py

app.py

@@ -96,31 +96,7 @@ temperature=0.5,
 model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',# it is possible that this model may be overloaded
 custom_role_conversions=None,
 )
-@tool
-def image_generator(prompt: str) -> bytes:
-    """Generate an image based on a text description.
-    Args:
-        prompt: Detailed description of the image you want to generate
-    Returns:
-        The image data as bytes that can be displayed
-    """
-    import requests
-    
-    API_URL = "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5"
-    
-    # Make API request without authentication headers
-    response = requests.post(
-        API_URL,
-        json={"inputs": prompt}
-    )
-    
-    # Check if the request was successful
-    if response.status_code != 200:
-        raise Exception(f"Error generating image: {response.text}")
-    
-    # Return the raw image bytes
-    return response.content
-    
+
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 search_tool = DuckDuckGoSearchTool()
@@ -129,7 +105,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer,enhanced_translate,image_generator,search_tool ], ## add your tools here (don't remove final answer)
+    tools=[final_answer,enhanced_translate,image_generation_tool,search_tool ], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,