Update app.py

app.py

@@ -9,9 +9,16 @@ import spaces
 import torch
 from diffusers import StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler
 
-DESCRIPTIONx = """## STABLE HAMSTER
+
+DESCRIPTIONx = """
+<div style="text-align:center">
+    <iframe src="https://prithivmlmods-hamster-static.static.hf.space/index.html" width="560" height="315" frameborder="0" allowfullscreen></iframe>
+</div>
+
+## STABLE HAMSTER
 """
 
+
 # Use environment variables for flexibility
 MODEL_ID = os.getenv("MODEL_REPO")
 MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "4096"))
@@ -112,24 +119,7 @@ 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():
@@ -235,7 +225,5 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
         api_name="run",
     )
 
-    gr.HTML(loading_js)
-
 if __name__ == "__main__":
     demo.queue(max_size=50).launch()