fixed image gen

thijsi123 · Feb 1, 2024 · dc5a41c · dc5a41c
1 parent 1c1dfe1
commit dc5a41c
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 3 deletions.
diff --git a/app/oobabooga - webui.py b/app/oobabooga - webui.py
@@ -735,7 +735,6 @@ def generate_character_avatar(
     )
     print(sd_prompt)
     sd_filter(nsfw_filter)
-    process_uploaded_image
     return image_generate(character_name,
                           sd_prompt,
                           input_none(negative_prompt),
@@ -766,9 +765,18 @@ def image_generate(character_name, prompt, negative_prompt):
     character_name = character_name.replace(" ", "_")
     os.makedirs(f"characters/{character_name}", exist_ok=True)
 
-    card_path = f"characters/{character_name}/{character_name}.png"
+    image_path = f"characters/{character_name}/{character_name}.png"
+
+    # Save the generated image
+    generated_image.save(image_path)
+
+    # Load the image back into a NumPy array
+    reloaded_image = Image.open(image_path)
+    reloaded_image_np = np.array(reloaded_image)
+
+    # Call process_uploaded_image
+    process_uploaded_image(reloaded_image_np)
 
-    generated_image.save(card_path)
     print("Generated character avatar")
     return generated_image
 

diff --git a/characters/Kaori/Kaori.card.png b/characters/Kaori/Kaori.card.png
diff --git a/characters/Kaori/Kaori.png b/characters/Kaori/Kaori.png
diff --git a/characters/uploaded_character/uploaded_character.png b/characters/uploaded_character/uploaded_character.png