22

app.py

@@ -65,7 +65,7 @@ def generate_lora(
     progress=gr.Progress(track_tqdm=True),
 ):
 
-    ulid = str(ULID()).lower()
+    ulid = str(ULID()).lower()[:12]
     print(f"ulid: {ulid}")
 
     # Load images 
@@ -88,7 +88,7 @@ def generate_lora(
 
     save_file(lora, lora_path)
 
-    return True
+    return lora_path, lora_url, True
 
 @spaces.GPU
 def generate_image(
@@ -140,9 +140,9 @@ 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 my_container:
+        with gr.Group(visible=has_lora) as imagen_container:
+            gr.Markdown("Your LoRA is ready! Now, try generating some images.")
             with gr.Row():
-                gr.Markdown("Your LoRA is ready! Now, try generating some images.")
                 with gr.Column():
                     prompt = gr.Textbox(
                         label="Prompt",
@@ -210,7 +210,7 @@ with gr.Blocks() as demo:
         inputs=[
             input_images
         ],
-        outputs=[lora_path, lora_download, has_lora],
+        outputs=[lora_path, lora_download, has_lora, imagen_container],
     )