Update app.py

app.py

@@ -43,7 +43,6 @@ pipe.load_lora_weights(hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8
 pipe.fuse_lora(lora_scale=0.125)
 pipe.to(device="cuda", dtype=torch.bfloat16)
 
-# Custom CSS
 css = """
 footer {display: none !important}
 .gradio-container {max-width: 1200px; margin: auto;}
@@ -66,6 +65,7 @@ footer {display: none !important}
     -webkit-background-clip: text;
     -webkit-text-fill-color: transparent;
 }
+/* 갤러리 컨테이너 CSS 수정 */
 .gallery-container {
     display: grid;
     grid-template-columns: repeat(auto-fill, minmax(150px, 1fr));
@@ -74,6 +74,7 @@ footer {display: none !important}
     background: rgba(255, 255, 255, 0.05);
     border-radius: 8px;
     margin-top: 10px;
+    width: 100%;
 }
 .gallery-image {
     width: 100%;
@@ -85,6 +86,10 @@ footer {display: none !important}
 .gallery-image:hover {
     transform: scale(1.05);
 }
+/* 갤러리와 출력 이미지 컨테이너 너비 통일 */
+.output-image, #gallery {
+    width: 100% !important;
+}
 """
 
 def save_image(image):
@@ -111,7 +116,8 @@ def load_gallery():
 with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
     gr.HTML('<div class="title">AI Image Generator</div>')
     gr.HTML('<div style="text-align: center; margin-bottom: 2em; color: #666;">Create stunning images from your descriptions</div>')
-    
+
+
     with gr.Row():
         with gr.Column(scale=3):
             prompt = gr.Textbox(
@@ -194,12 +200,12 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
                     </div>
                 </div>
             """)
-        
-        with gr.Column(scale=4):
-            # Current generated image
-            output = gr.Image(label="Generated Image")
             
-            # Gallery of generated images
+        with gr.Column(scale=4):
+        # Current generated image
+            output = gr.Image(label="Generated Image", elem_id="output-image", elem_classes=["output-image"])
+    
+        # Gallery of generated images
             gallery = gr.Gallery(
                 label="Generated Images Gallery",
                 show_label=True,
@@ -207,8 +213,11 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
                 columns=[4],
                 rows=[2],
                 height="auto",
-                object_fit="contain"
-            )
+                object_fit="contain",
+                elem_classes=["gallery-container"]
+            )    
+                
+
             
             # Load existing gallery images on startup
             gallery.value = load_gallery()