Update app.py

app.py

@@ -1514,16 +1514,13 @@ with gr.Blocks(theme='Pijush2023/scikit-learn-pijush') as demo:
             # refresh_button = gr.Button("Refresh Images")
             # refresh_button.click(fn=update_images, inputs=None, outputs=[image_output_1, image_output_2, image_output_3])
           
-            # Displaying the images generated using Flux API
-            image_output_1 = gr.Image(label="Image 1", elem_id="flux_image_1", width=400, height=400)
-            image_output_2 = gr.Image(label="Image 2", elem_id="flux_image_2", width=400, height=400)
-            image_output_3 = gr.Image(label="Image 3", elem_id="flux_image_3", width=400, height=400)
-
             # Call update_images during the initial load to display images when the interface appears
             initial_images = update_images()
-            image_output_1.update(value=initial_images[0])
-            image_output_2.update(value=initial_images[1])
-            image_output_3.update(value=initial_images[2])
+            
+            # Displaying the images generated using Flux API directly
+            image_output_1 = gr.Image(value=initial_images[0], label="Image 1", elem_id="flux_image_1", width=400, height=400)
+            image_output_2 = gr.Image(value=initial_images[1], label="Image 2", elem_id="flux_image_2", width=400, height=400)
+            image_output_3 = gr.Image(value=initial_images[2], label="Image 3", elem_id="flux_image_3", width=400, height=400)
     
             # Refresh button to update images
             refresh_button = gr.Button("Refresh Images")