Update app.py

app.py

@@ -63,32 +63,17 @@ def generate_prompts_for_object(object_name):
         "future": f"Show a futuristic version of a {object_name}, by predicting advanced features and futuristic design."
     }
 
-'''
+
 image_generation_tool = Tool.from_space(
-    "KingNish/Realtime-FLUX",
-    api_name="/predict",  # Optional if there's only one endpoint
-    name="image_generator",
-    description="Generate an image from a prompt"
-)
-'''
-import requests
-from smolagents import Tool
-
-def flux_proxy(user_prompt: str):
-    # Call the Hugging Face Space API directly
-    url = "https://black-forest-labs-flux-1-schnell.hf.space/run/infer"
-    headers = {"Authorization": f"Bearer {token}"}
-    response = requests.post(url, headers=headers, json={"data": [user_prompt]})
-    response.raise_for_status()
-    output_url = response.json()["data"][0]  # Usually a URL or base64 image
-    return output_url  # Adjust this based on your agent's expectations
-
-image_generation_tool = Tool.from_function(
-    fn=flux_proxy,
+    #"KingNish/Realtime-FLUX",
+    "black-forest-labs/FLUX.1-schnell",
+    num_inference_steps=4,
+	api_name="/infer"
     name="image_generator",
     description="Generate an image from a prompt"
 )
 
+
 # =========================================================
 # Tool and Agent Initialization
 # =========================================================