22

app.py

@@ -88,7 +88,7 @@ def generate_lora(
 
     save_file(lora, lora_path)
 
-    return lora_path, lora_path, gr.Column(visible=True)
+    return lora_path, lora_path, gr.update(visible=True), gr.update(visible=True)
 
 @spaces.GPU
 def generate_image(
@@ -125,7 +125,6 @@ h3{
 
 with open('examples/0_examples.json', 'r') as file: examples = json.load(file)
 with gr.Blocks() as demo:
-    has_lora = gr.State(False)
     with gr.Column(elem_id="col-container"):
         with gr.Column():
             gr.HTML(read_file("static/header.html"))
@@ -143,8 +142,8 @@ with gr.Blocks() as demo:
 
             with gr.Column():
                 lora_path = gr.Textbox(label="Generated LoRA path",lines=2, interactive=False)
-                lora_download = gr.DownloadButton(label=f"Download LoRA", visible=has_lora)
-        with gr.Accordion("Generate Image", open=False):
+                lora_download = gr.DownloadButton(label=f"Download LoRA", visible=False)
+        with gr.Column(visible=False) as imagen_container:
             gr.Markdown("### After your LoRA is ready, you can try generate image here.")
             with gr.Row():
                 with gr.Column():
@@ -212,7 +211,7 @@ with gr.Blocks() as demo:
         inputs=[
             input_images
         ],
-        outputs=[lora_path, lora_download],
+        outputs=[lora_path, lora_download, imagen_container],
     )