Update app.py

app.py

@@ -6,6 +6,7 @@ from safetensors.torch import load_file
 from share_btn import community_icon_html, loading_icon_html, share_js
 from uuid import uuid4
 from pathlib import Path
+from PIL import Image
 import torch
 import json
 import random
@@ -127,7 +128,7 @@ def shuffle_images():
 def save_preferences(lora_1_id, lora_1_scale, lora_2_id, lora_2_scale, prompt, generated_image, thumbs_direction, seed):
     image_path = IMAGE_DATASET_DIR / f"{uuid4()}.png"
     with scheduler.lock:
-        generated_image.save(image_path)
+        Image.fromarray(image_array).save(image_path)
         with IMAGE_JSONL_PATH.open("a") as f:
             json.dump({"prompt": prompt, "file_name":image_path.name, "lora_1_id": lora_2_id, "lora_1_scale": lora_1_scale, "lora_2_id": lora_2_id, "lora_2_scale": lora_2_scale, "thumbs_direction": thumbs_direction, "seed": seed}, f)
             f.write("\n")
@@ -144,13 +145,13 @@ with gr.Blocks(css=css) as demo:
   with gr.Column():
     with gr.Column(min_width=10, scale=16, elem_classes="plus_column"):
         with gr.Row():
-            with gr.Column(min_width=10, scale=3, elem_classes="random_column"):
+            with gr.Column(min_width=10, scale=4, elem_classes="random_column"):
               lora_1 = gr.Image(interactive=False, height=150, elem_classes="selected_random", elem_id="randomLoRA_1", show_share_button=False, show_download_button=False)
               lora_1_id = gr.Textbox(visible=False, elem_id="random_lora_1_id")
               lora_1_prompt = gr.Markdown(visible=False)
             with gr.Column(min_width=10, scale=1, elem_classes="plus_column"):
               plus = gr.HTML("+", elem_classes="plus_button")
-            with gr.Column(min_width=10, scale=3, elem_classes="random_column"):
+            with gr.Column(min_width=10, scale=4, elem_classes="random_column"):
               lora_2 = gr.Image(interactive=False, height=150, elem_classes="selected_random", elem_id="randomLoRA_2", show_share_button=False, show_download_button=False)
               lora_2_id = gr.Textbox(visible=False, elem_id="random_lora_2_id")
               lora_2_prompt = gr.Markdown(visible=False)