Update app.py

app.py

@@ -128,8 +128,10 @@ class TextToImageTool(Tool):
 '''
 image_generation_tool = Tool.from_space(
     "black-forest-labs/FLUX.1-schnell",
+    api_name="/infer",  # Optional if there's only one endpoint
     name="image_generator",
-    description="Generate an image from a prompt")
+    description="Generate an image from a prompt"
+)
 
 
 # =========================================================
@@ -193,10 +195,10 @@ def generate_object_history(object_name):
 
     gif_path = f"{object_name}_evolution.gif"
     images[0].save(gif_path, save_all=True, append_images=images[1:], duration=1000, loop=0)
-    #return [(f"{object_name}_past.png", labels["past"]),
-    #        (f"{object_name}_present.png", labels["present"]),
-    #        (f"{object_name}_future.png", labels["future"])], gif_path
-    return images, gif_path
+    return [(f"{object_name}_past.png", labels["past"]),
+            (f"{object_name}_present.png", labels["present"]),
+            (f"{object_name}_future.png", labels["future"])], gif_path
+    #return images, gif_path
 
 # =========================================================
 # Gradio Interface
@@ -221,8 +223,8 @@ def create_gradio_interface():
             with gr.Column():
                 object_name_input = gr.Textbox(label="Enter an object name", placeholder="e.g. bicycle, car, phone")
                 generate_button = gr.Button("Generate Evolution")
-                #image_gallery = gr.Gallery(label="Generated Images", columns=3, rows=1, value=default_images)
-                image_gallery = gr.Gallery(label="Generated Images", columns=3, rows=1, type="filepath")
+                image_gallery = gr.Gallery(label="Generated Images", columns=3, rows=1, value=default_images)
+                #image_gallery = gr.Gallery(label="Generated Images", columns=3, rows=1, type="filepath")
                 gif_output = gr.Image(label="Generated GIF", value=default_gif_path)
 
         generate_button.click(fn=generate_object_history, inputs=[object_name_input], outputs=[image_gallery, gif_output])