22

app.py

@@ -91,21 +91,9 @@ def generate_lora(
     ulid = str(ULID()).lower()[:12]
     print(f"ulid: {ulid}")
 
-    # print("before: ", input_images)
-    # input_images = [item[0] for item in input_images]
     input_images = [Image.open(filepath) for filepath, _ in input_images]
-    # print("after: ", input_images)
     if not input_images:
         print("images are empty.")
-    # Load images 
-    # images = [
-    #     Image.open("examples/style/1/0.jpg"),
-    #     Image.open("examples/style/1/1.jpg"),
-    #     Image.open("examples/style/1/2.jpg"),
-    #     Image.open("examples/style/1/3.jpg"),
-    #     Image.open("examples/style/1/4.jpg"),
-    # ]
-
 
     # Model inference
     with torch.no_grad():
@@ -117,7 +105,7 @@ def generate_lora(
 
     save_file(lora, lora_path)
 
-    return lora_name, gr.update(visible=True, value=lora_path), gr.update(visible=True)
+    return lora_name, gr.update(visible=True, value=lora_path), gr.update(interactive=True)
 
 @spaces.GPU
 def generate_image(
@@ -199,7 +187,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)
-        with gr.Column(visible=False, elem_id='imagen-container') as imagen_container:
+        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():
                 with gr.Column():
@@ -208,11 +196,11 @@ with gr.Blocks() as demo:
                         show_label=False,
                         lines=2,
                         placeholder="Enter your prompt",
-                        value="a man in a fishing boat. high quality, detailed",
+                        value="a man in a fishing boat.",
                         container=False,
                     )
                     
-                    imagen_button = gr.Button("Generate Image", variant="primary")    
+                    imagen_button = gr.Button("Generate Image", variant="primary", interactive=False)    
                     with gr.Accordion("Advanced Settings", open=False):
                         negative_prompt = gr.Textbox(
                             label="Negative prompt",
@@ -264,7 +252,7 @@ with gr.Blocks() as demo:
                 with gr.Column():
                     output_image = gr.Image(label="Generated image", show_label=False)
                     
-        # gr.Examples(examples=examples, inputs=[input_image])
+        gr.Examples(examples=examples, inputs=[input_images])
         gr.Markdown(read_file("static/footer.md"))
 
     lora_button.click(
@@ -272,7 +260,7 @@ with gr.Blocks() as demo:
         inputs=[
             input_images
         ],
-        outputs=[lora_name, lora_download, imagen_container],
+        outputs=[lora_name, lora_download, imagen_button],
     )
     imagen_button.click(
         fn=generate_image,

examples/0_examples.json

@@ -1,8 +1,9 @@
 [
-	"Saitama, punching",
-	"Levi Ackerman, kick",
-	"Goku, powering up",
-	"Light Yagami, writing",
-	"Naruto, eating ramen",
-	"Luffy, grinning"
+	[
+		"examples/style/1/0.jpg",
+		"examples/style/1/1.jpg",
+		"examples/style/1/2.jpg",
+		"examples/style/1/3.jpg",
+		"examples/style/1/4.jpg"
+	]
 ]

static/header.html

@@ -6,6 +6,6 @@
         <p>
             Generate LoRA from a few images.
             <br>
-            If you like my work, please support me by visiting <a href="https://aisudo.com/" target="_blank">AiSudo</a> 😊 
+            Demo by <a href="https://aisudo.com/" target="_blank">AiSudo</a> 😊 
     </div>
 </div>