Create app.py

app.py

@@ -38,8 +38,6 @@ torch.backends.cudnn.benchmark = False
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
-# Add a new global variable to store the image history
-image_history = []
 
 def load_pipeline(model_name):
     vae = AutoencoderKL.from_pretrained(
@@ -66,6 +64,7 @@ def load_pipeline(model_name):
     pipe.to(device)
     return pipe
 
+
 @spaces.GPU
 def generate(
     prompt: str,
@@ -160,17 +159,7 @@ def generate(
                 filepath = utils.save_image(image, metadata, OUTPUT_DIR)
                 logger.info(f"Image saved as {filepath} with metadata")
 
-        # Add the generated image and metadata to the history
-        for image in images:
-            thumbnail = image.copy()
-            thumbnail.thumbnail((256, 256))
-            image_history.insert(0, {
-                "image": thumbnail,
-                "prompt": prompt,
-                "metadata": metadata
-            })
-
-        return images, metadata, gr.update(value=image_history)
+        return images, metadata
     except Exception as e:
         logger.exception(f"An error occurred: {e}")
         raise
@@ -180,6 +169,7 @@ def generate(
         pipe.scheduler = backup_scheduler
         utils.free_memory()
 
+
 if torch.cuda.is_available():
     pipe = load_pipeline(MODEL)
     logger.info("Loaded on Device!")
@@ -202,35 +192,24 @@ with gr.Blocks(css="style.css") as demo:
     )
     with gr.Group():
         with gr.Row():
-            with gr.Column(scale=2):
-                prompt = gr.Text(
-                    label="Prompt",
-                    show_label=False,
-                    max_lines=5,
-                    placeholder="Enter your prompt",
-                    container=False,
-                )
-                run_button = gr.Button(
-                    "Generate", 
-                    variant="primary", 
-                    scale=0
-                )
-                result = gr.Gallery(
-                    label="Result", 
-                    columns=1, 
-                    preview=True, 
-                    show_label=False
-                )
-            
-            with gr.Column(scale=1):
-                history = gr.Gallery(
-                    label="Generation History",
-                    show_label=True,
-                    elem_id="history",
-                    columns=2,
-                    height=800,
-                )
-
+            prompt = gr.Text(
+                label="Prompt",
+                show_label=False,
+                max_lines=5,
+                placeholder="Enter your prompt",
+                container=False,
+            )
+            run_button = gr.Button(
+                "Generate", 
+                variant="primary", 
+                scale=0
+            )
+        result = gr.Gallery(
+            label="Result", 
+            columns=1, 
+            preview=True, 
+            show_label=False
+        )
     with gr.Accordion(label="Advanced Settings", open=False):
         negative_prompt = gr.Text(
             label="Negative Prompt",
@@ -311,7 +290,7 @@ with gr.Blocks(css="style.css") as demo:
     gr.Examples(
         examples=config.examples,
         inputs=prompt,
-        outputs=[result, gr_metadata, history],
+        outputs=[result, gr_metadata],
         fn=lambda *args, **kwargs: generate(*args, use_upscaler=True, **kwargs),
         cache_examples=CACHE_EXAMPLES,
     )
@@ -354,7 +333,7 @@ with gr.Blocks(css="style.css") as demo:
     ).then(
         fn=generate,
         inputs=inputs,
-        outputs=[result, gr_metadata, history],
+        outputs=result,
         api_name="run",
     )
     negative_prompt.submit(
@@ -366,7 +345,7 @@ with gr.Blocks(css="style.css") as demo:
     ).then(
         fn=generate,
         inputs=inputs,
-        outputs=[result, gr_metadata, history],
+        outputs=result,
         api_name=False,
     )
     run_button.click(
@@ -378,8 +357,7 @@ with gr.Blocks(css="style.css") as demo:
     ).then(
         fn=generate,
         inputs=inputs,
-        outputs=[result, gr_metadata, history],
+        outputs=[result, gr_metadata],
         api_name=False,
     )
-
 demo.queue(max_size=20).launch(debug=IS_COLAB, share=IS_COLAB)