Update app.py

app.py

@@ -67,7 +67,6 @@ def generate_prompts_for_object(object_name):
 image_generation_tool = Tool.from_space(
     #"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"
@@ -114,7 +113,7 @@ def generate_object_history(object_name):
             )
             
             if isinstance(result, (list, tuple)):
-            result = result[0]
+                result = result[0]
             
             image = result.to_raw()
         except Exception as e: