22

app.py

@@ -83,10 +83,8 @@ def generate_lora(
         embs = QwenImageUnit_Image2LoRAEncode().process(pipe, image2lora_images=images)
         lora = QwenImageUnit_Image2LoRADecode().process(pipe, **embs)["lora"]
 
-    lora_path = f"loras/{ulid}.safetensors"
-    # lora_url = f"{URL_PUBLIC}/{lora_path}"
-
-    # print(lora_url)
+    lora_name = f"{ulid}.safetensors"
+    lora_path = f"loras/{lora_name}"
 
     save_file(lora, lora_path)
 
@@ -117,6 +115,10 @@ css = """
     margin: 0 auto;
     max-width: 960px;
 }
+h3{
+    text-align: center;
+    display:block;
+}
 """
 
 
@@ -142,8 +144,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.Column(visible=False) as imagen_container:
-            gr.Markdown("Your LoRA is ready! Now, try generating some images.")
+        with gr.Accordion("Generate Image", open=False):
+            gr.Markdown("### After your LoRA is ready, you can try generate image her.")
             with gr.Row():
                 with gr.Column():
                     prompt = gr.Textbox(
@@ -155,9 +157,7 @@ with gr.Blocks() as demo:
                         # container=False,
                     )
                     
-                    imagen_button = gr.Button("Generate Image", variant="primary")
-
-                    
+                    imagen_button = gr.Button("Generate Image", variant="primary")    
                     with gr.Accordion("Advanced Settings", open=False):
                         
                         negative_prompt = gr.Textbox(

static/header.html

@@ -1,6 +1,6 @@
 <div style="text-align: center; max-width: 600px; margin: 0 auto;">
     <h1>
-        Z-Image Turbo (ZIT) - Upscaler
+        Qwen Image to LoRA
     </h1>
     <div class="grid-container" >
         <p>