Update app.py

app.py

@@ -10,13 +10,23 @@ import torch
 from diffusers import StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler
 
 # Load the HTML content
-html_file_url = "https://prithivmlmods-hamster-static.static.hf.space/index.html"
-html_content = f'<iframe src="{html_file_url}" style="width:100%; height:200px; border:none;></iframe>'
+#html_file_url = "https://prithivmlmods-hamster-static.static.hf.space/index.html"
+#html_content = f'<iframe src="{html_file_url}" style="width:100%; height:200px; border:none;></iframe>'
 
 
-DESCRIPTIONx = """## STABLE HAMSTER
+#DESCRIPTIONx = """## STABLE HAMSTER
+#"""
+
+# Load the uploaded image content
+html_image_path = "https://prithivmlmods-hamster-static.static.hf.space/index.html"
+
+DESCRIPTIONx = """<div style="display: flex; align-items: center; justify-content: space-between;">
+    <h1 style="margin: 0;">STABLE HAMSTER</h1>
+    <img src="https://prithivmlmods-hamster-static.static.hf.space/index.html" style="width: 100px; height: auto; margin-left: 20px;">
+</div>
 """
 
+
 # Use environment variables for flexibility
 MODEL_ID = os.getenv("MODEL_REPO")
 MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "4096"))
@@ -118,7 +128,7 @@ footer {
 '''
 
 with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
-    gr.HTML(html_content)# Add the HTML content here
+    #gr.HTML(html_content)# Add the HTML content here
     gr.Markdown(DESCRIPTIONx)
     with gr.Group():
         with gr.Row():