22

app.py

@@ -105,7 +105,7 @@ def generate_lora(
 
     save_file(lora, lora_path)
 
-    return lora_name, gr.update(visible=True, value=lora_path), gr.update(interactive=True)
+    return lora_name, gr.update(interactive=True, value=lora_path), gr.update(interactive=True)
 
 @spaces.GPU
 def generate_image(
@@ -158,9 +158,7 @@ h3{
     text-align: center;
     display:block;
 }
-#imagen-container {
-    padding: 12px;
-}
+
 """
 
 
@@ -186,7 +184,7 @@ with gr.Blocks() as demo:
 
             with gr.Column():
                 lora_name = gr.Textbox(label="Generated LoRA path",lines=2, interactive=False)
-                lora_download = gr.DownloadButton(label=f"Download LoRA", visible=False)
+                lora_download = gr.DownloadButton(label=f"Download LoRA", interactive=False)
         with gr.Column(elem_id='imagen-container') as imagen_container:
             gr.Markdown("### After your LoRA is ready, you can try generate image here.")
             with gr.Row():