Update app.py

app.py

@@ -3,7 +3,6 @@ import threading
 import os
 import torch
 
-
 os.environ["OMP_NUM_THREADS"] = str(os.cpu_count())
 torch.set_num_threads(os.cpu_count())
 
@@ -33,13 +32,14 @@ def generate_images(text, selected_model):
 
     return results
 
-
 def stop_generation():
+    """Stops the ongoing image generation by setting the stop_event flag."""
     stop_event.set()
+    return ["Generation stopped."] * 3
 
 with gr.Blocks() as interface:
     gr.Markdown(
-        "### ⚠ Sorry for the inconvenience. The model is currently running on the CPU, which might affect performance. We appreciate your understanding."
+        "### ⚠ Sorry for the inconvenience. The Space is currently running on the CPU, which might affect performance. We appreciate your understanding."
     )
     
     text_input = gr.Textbox(label="Type here your imagination:", placeholder="Type your prompt...")
@@ -50,7 +50,7 @@ with gr.Blocks() as interface:
     )
     
     with gr.Row():
-        generate_button = gr.Button("Generate 3 Images")
+        generate_button = gr.Button("Generate 3 Images 🎨")
         stop_button = gr.Button("Stop Image Generation")
     
     with gr.Row():
@@ -59,6 +59,6 @@ with gr.Blocks() as interface:
         output3 = gr.Image(label="Generated Image 3")
     
     generate_button.click(generate_images, inputs=[text_input, model_selector], outputs=[output1, output2, output3])
-    stop_button.click(stop_generation, inputs=[], outputs=[])
+    stop_button.click(stop_generation, inputs=[], outputs=[output1, output2, output3])
 
 interface.launch()