22

app.py

@@ -88,7 +88,7 @@ def generate_lora(
 
     save_file(lora, lora_path)
 
-    return lora_path, lora_url, True
+    return lora_path, lora_url, gr.Column(visible=True)
 
 @spaces.GPU
 def generate_image(
@@ -140,7 +140,7 @@ 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.Group(visible=has_lora) as imagen_container:
+        with gr.Column(visible=False) as imagen_container:
             gr.Markdown("Your LoRA is ready! Now, try generating some images.")
             with gr.Row():
                 with gr.Column():
@@ -210,7 +210,7 @@ with gr.Blocks() as demo:
         inputs=[
             input_images
         ],
-        outputs=[lora_path, lora_download, has_lora, imagen_container],
+        outputs=[lora_path, lora_download, imagen_container],
     )