Update app.py

app.py

@@ -64,6 +64,14 @@ def generate(
     num_images: int = 1,  # Number of images to generate
     progress=gr.Progress(track_tqdm=True),
 ):
+    # Show the loading animation
+    loading_html = """
+    <div id="loading-animation">
+        <iframe src="https://prithivmlmods-hamster-static.static.hf.space/index.html" width="100%" height="200" style="border:none;"></iframe>
+    </div>
+    """
+    gr.HTML.update(loading_html, visible=True)
+    
     seed = int(randomize_seed_fn(seed, randomize_seed))
     generator = torch.Generator(device=device).manual_seed(seed)
 
@@ -93,6 +101,10 @@ def generate(
         images.extend(pipe(**batch_options).images)
 
     image_paths = [save_image(img) for img in images]
+    
+    # Hide the loading animation
+    gr.HTML.update("", visible=False)
+    
     return image_paths, seed
 
 examples = [
@@ -124,7 +136,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) 
+        result = gr.Gallery(label="Result", columns=1, show_label=False)
         loading_animation = gr.HTML(
             """
             <div id="loading-animation" style="display:none;">
@@ -202,22 +214,20 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
         api_name=False,
     )
 
-    def start_loading():
-        return gr.update(visible=True), gr.update(visible=False)
-
-    def stop_loading():
-        return gr.update(visible=False), gr.update(visible=True)
-
-    prompt.submit(start_loading, [], [loading_animation, result], queue=False)
-    run_button.click(start_loading, [], [loading_animation, result], queue=False)
+    prompt.submit(fn=generate, inputs=[
+            prompt,
+            negative_prompt,
+            use_negative_prompt,
+            seed,
+            width,
+            height,
+            guidance_scale,
+            num_inference_steps,
+            randomize_seed,
+            num_images
+        ], outputs=[result, seed], queue=True)
 
-    gr.on(
-        triggers=[
-            prompt.submit,
-            run_button.click,
-        ],
-        fn=generate,
-        inputs=[
+    run_button.click(fn=generate, inputs=[
             prompt,
             negative_prompt,
             use_negative_prompt,
@@ -228,11 +238,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
             num_inference_steps,
             randomize_seed,
             num_images
-        ],
-        outputs=[result, seed],
-        api_name="run",
-        postprocess_fn=stop_loading,
-    )
+        ], outputs=[result, seed], queue=True)
 
 if __name__ == "__main__":
     demo.queue(max_size=50).launch()