Update app.py

app.py

@@ -1517,8 +1517,8 @@ with gr.Blocks(theme='Pijush2023/scikit-learn-pijush') as demo:
           
             # 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)
+            # 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)
 
             # Refresh button to update images
             refresh_button = gr.Button("Refresh Images")