Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import numpy as np
-import spaces # This is a special module for Hugging Face Spaces, not needed for local execution
+import spaces
 import torch
 import random
 import json
@@ -123,6 +123,8 @@ def infer_with_lora(input_image, prompt, selected_index, seed=0, guidance_scale=
 
 # CSS styling
 css = """
+#gen_btn{height: 100%}
+#gen_column{align-self: stretch}
 #main_app {
     display: flex;
     gap: 20px;
@@ -165,15 +167,13 @@ css = """
 """
 
 # Create Gradio interface
-with gr.Blocks(css=css, theme=gr.themes.Ocean(font=[gr.themes.GoogleFont("Lexend Deca"), "sans-serif"])) as demo:
+with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
     gr_flux_loras = gr.State(value=flux_loras_raw)
     
     title = gr.HTML(
-        """<h1><img src="https://huggingface.co/spaces/kontext-community/FLUX.1-Kontext-portrait/resolve/main/dora_kontext.png" alt="LoRA"> Kontext-Style LoRA Explorer</h1>""",
+        """<h1>FLUX Kontext Super LoRAs🖖</h1>""",
         elem_id="title",
     )
-    gr.Markdown("A demo for the style LoRAs from the [Kontext-Style](https://huggingface.co/Kontext-Style) 🤗")
-    
     selected_state = gr.State(value=None)
     # The following states are no longer used by the simplified logic but kept for component structure
     custom_loaded_lora = gr.State(value=None)
@@ -216,7 +216,7 @@ with gr.Blocks(css=css, theme=gr.themes.Ocean(font=[gr.themes.GoogleFont("Lexend
                     placeholder="opt - describe the person/subject, e.g. 'a man with glasses and a beard'",
                     elem_id="prompt"
                 )
-                run_button = gr.Button("Generate", elem_id="run_button")
+                run_button = gr.Button("Generate", variant="primary", elem_id="gen_btn")
             
             result = gr.Image(label="Generated Image", interactive=False, height=512)