Update app.py

app.py

@@ -93,8 +93,7 @@ def generate(
         images.extend(pipe(**batch_options).images)
 
     image_paths = [save_image(img) for img in images]
-    
-    return image_paths, seed, gr.update(visible=False)
+    return image_paths, seed
 
 examples = [
     "a cat eating a piece of cheese",
@@ -113,7 +112,24 @@ footer {
 }
 '''
 
+loading_html = """
+<div id="loading" style="display: flex; justify-content: center; align-items: center; height: 100vh;">
+    <iframe src="https://prithivmlmods-hamster-static.static.hf.space/index.html" style="border: none; width: 100%; height: 100%;"></iframe>
+</div>
+<div id="app" style="display: none;">
+"""
+
+loading_js = """
+<script>
+    setTimeout(() => {
+        document.getElementById('loading').style.display = 'none';
+        document.getElementById('app').style.display = 'block';
+    }, 5000);  // 5000 milliseconds = 5 seconds
+</script>
+"""
+
 with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
+    gr.HTML(loading_html)
     gr.Markdown(DESCRIPTIONx)
     with gr.Group():
         with gr.Row():
@@ -125,14 +141,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
                 container=False,
             )
             run_button = gr.Button("Run", scale=0)
-        result = gr.Gallery(label="Result", columns=1, show_label=False)
-        loading_animation = gr.HTML(
-            """
-            <div id="loading-animation" style="display:none;">
-                <iframe src="https://prithivmlmods-hamster-static.static.hf.space/index.html" width="100%" height="200" style="border:none;"></iframe>
-            </div>
-            """
-        )
+        result = gr.Gallery(label="Result", columns=1, show_label=False) 
     with gr.Accordion("Advanced options", open=False):
         num_images = gr.Slider(
             label="Number of Images",
@@ -203,7 +212,14 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
         api_name=False,
     )
 
-    prompt.submit(fn=generate, inputs=[
+    gr.on(
+        triggers=[
+            prompt.submit,
+            negative_prompt.submit,
+            run_button.click,
+        ],
+        fn=generate,
+        inputs=[
             prompt,
             negative_prompt,
             use_negative_prompt,
@@ -214,20 +230,12 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
             num_inference_steps,
             randomize_seed,
             num_images
-        ], outputs=[result, seed, loading_animation], queue=True)
+        ],
+        outputs=[result, seed],
+        api_name="run",
+    )
 
-    run_button.click(fn=generate, inputs=[
-            prompt,
-            negative_prompt,
-            use_negative_prompt,
-            seed,
-            width,
-            height,
-            guidance_scale,
-            num_inference_steps,
-            randomize_seed,
-            num_images
-        ], outputs=[result, seed, loading_animation], queue=True)
+    gr.HTML(loading_js)
 
 if __name__ == "__main__":
     demo.queue(max_size=50).launch()