Update app.py

app.py

@@ -1,9 +1,8 @@
-import spaces # type: ignore
+import spaces  # type: ignore
 import os
-import time  # noqa: F401
 import uuid
 from PIL import Image
-import gradio as gr # type: ignore
+import gradio as gr
 import numpy as np
 import random
 import torch
@@ -62,9 +61,8 @@ def infer(
 
 
 examples = [
-    "a tiny astronaut hatching from an egg on the moon",
-    "a cat holding a sign that says hello world",
-    "an anime illustration of a wiener schnitzel",
+    "a cat holding a sign that says flux.1 is great",
+    "an old man holding a sign that says Increase Zero-GPU Limit",
 ]
 
 css = """
@@ -82,13 +80,14 @@ with gr.Blocks(css=css) as demo:
         """)
 
         with gr.Row(equal_height=False):
-            prompt = gr.TextArea(
-                label="Prompt",
-                show_label=False,
-                lines=3,
-                placeholder="Enter your prompt",
-                container=False,
-            )
+            with gr.Column():
+                prompt = gr.TextArea(
+                    label="Prompt",
+                    show_label=False,
+                    lines=3,
+                    placeholder="Enter your prompt",
+                    container=False,
+                )
 
             run_button = gr.Button("Run", variant="primary", scale=0)
 
@@ -147,17 +146,20 @@ with gr.Blocks(css=css) as demo:
                     value=28,
                 )
 
-        # gr.Examples(
-        #     examples=examples,
-        #     fn=infer,
-        #     inputs=[prompt],
-        #     outputs=[result, seed],
-        #     cache_examples="lazy",
-        #     visible=False,
-        # )
-
-    run_button.click(fn=lambda: gr.update(visible=False),inputs=None, outputs=download)
-    
+        gr.Examples(
+            examples=examples,
+            fn=infer,
+            inputs=[prompt],
+            outputs=[result, download, seed],
+            cache_examples="lazy",
+        )
+
+    gr.on(
+        triggers=[run_button.click],
+        fn=lambda: gr.update(visible=False),
+        outputs=download,
+        api_name=False,
+    )
     gr.on(
         triggers=[run_button.click, prompt.submit],
         fn=infer,
@@ -173,25 +175,6 @@ with gr.Blocks(css=css) as demo:
         outputs=[result, download, seed],
     )
 
-# def delete_packed_tensors(
-#     startup_time: int = time.time_ns(),
-#     zerogpu_offload_dir: str = "/home/user/.zerogpu/tensors",
-# ) -> None:
-#     entries = os.scandir(zerogpu_offload_dir)
-#     for entry in entries:
-#         print(f"Found entry {entry.path}")
-#         if entry.stat().st_atime_ns < startup_time:
-#             try:
-#                 file_path = entry.path
-#                 os.remove(file_path)
-#                 print(f"File deleted successfully: {file_path}")
-#             except FileNotFoundError:
-#                 print(f"File not found: {file_path}")
-#             except OSError as e:
-#                 print(f"Error deleting file: {e}")
-
-
 if __name__ == "__main__":
-    # delete_packed_tensors()
     os.makedirs("images", exist_ok=True)
     demo.launch()