rearn

check_app.py

@@ -21,19 +21,8 @@ import os
 from datetime import datetime
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
-gallery : gr.Gallery = None
-
-def initialize_components():
-    global gallery
     
-    gallery = gr.Gallery(
-    label="Image Gallery",
-    show_label=True,
-    columns=4,
-    rows=3,
-    height=600,
-    object_fit="contain"
-    )
+
     
 class ProgressPipeline(DiffusionPipeline):
     def __init__(self, original_pipeline):
@@ -274,15 +263,24 @@ def main():
                 output = gr.Textbox(label="Status")
                 img = gr.Image(label=model_name, height=300)
 
-                button.click(fn=create_pipeline_logic, inputs=[prompt_text, gr.Text(value= model_name,visible=False), negative_prompt,
-                        seed,
-                        randomize_seed,
-                        width,
-                        height,
-                        guidance_scale,
-                        num_inference_steps], outputs=[output, img, gallery])
 
-                gallery.render()
+
+                gallery = gr.Gallery(
+                    label="Image Gallery",
+                    show_label=True,
+                    columns=4,
+                    rows=3,
+                    height=600,
+                    object_fit="contain"
+                    )
+
+                button.click(fn=create_pipeline_logic, inputs=[prompt_text, gr.Text(value= model_name,visible=False), negative_prompt,
+                    seed,
+                    randomize_seed,
+                    width,
+                    height,
+                    guidance_scale,
+                    num_inference_steps], outputs=[output, img, gallery])