Update app.py

app.py

@@ -64,14 +64,6 @@ 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)
 
@@ -102,10 +94,7 @@ def generate(
 
     image_paths = [save_image(img) for img in images]
     
-    # Hide the loading animation
-    gr.HTML.update("", visible=False)
-    
-    return image_paths, seed
+    return image_paths, seed, gr.update(visible=False)
 
 examples = [
     "a cat eating a piece of cheese",
@@ -225,7 +214,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
             num_inference_steps,
             randomize_seed,
             num_images
-        ], outputs=[result, seed], queue=True)
+        ], outputs=[result, seed, loading_animation], queue=True)
 
     run_button.click(fn=generate, inputs=[
             prompt,
@@ -238,7 +227,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
             num_inference_steps,
             randomize_seed,
             num_images
-        ], outputs=[result, seed], queue=True)
+        ], outputs=[result, seed, loading_animation], queue=True)
 
 if __name__ == "__main__":
     demo.queue(max_size=50).launch()