Imagen examples

0x7o · May 31, 2023 · 1c923ea · 1c923ea
1 parent efe0a82
commit 1c923ea
Show file tree

Hide file tree

Showing 17 changed files with 3 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,5 +1,6 @@
 # Byte-compiled / optimized / DLL files
 __pycache__/
+.idea/
 *.py[cod]
 *$py.class
 

diff --git a/...3d_render_of_a_cute_tropical_fish_in_an_aquarium_on_a_dark_blue_background1.png b/...3d_render_of_a_cute_tropical_fish_in_an_aquarium_on_a_dark_blue_background1.png
diff --git a/...3d_render_of_a_cute_tropical_fish_in_an_aquarium_on_a_dark_blue_background2.png b/...3d_render_of_a_cute_tropical_fish_in_an_aquarium_on_a_dark_blue_background2.png
diff --git a/examples/3d_render_of_a_pink_balloon_dog_in_a_violet_room1.png b/examples/3d_render_of_a_pink_balloon_dog_in_a_violet_room1.png
diff --git a/examples/3d_render_of_a_pink_balloon_dog_in_a_violet_room2.png b/examples/3d_render_of_a_pink_balloon_dog_in_a_violet_room2.png
diff --git a/...s/a_photo_of_a_samoyed_dog_with_its_tongue_out_hugging_a_white_siamese_cat1.png b/...s/a_photo_of_a_samoyed_dog_with_its_tongue_out_hugging_a_white_siamese_cat1.png
diff --git a/...s/a_photo_of_a_samoyed_dog_with_its_tongue_out_hugging_a_white_siamese_cat2.png b/...s/a_photo_of_a_samoyed_dog_with_its_tongue_out_hugging_a_white_siamese_cat2.png
diff --git a/examples/a_plush_toy_robot_sitting_against_a_yellow_wall1.png b/examples/a_plush_toy_robot_sitting_against_a_yellow_wall1.png
diff --git a/examples/a_plush_toy_robot_sitting_against_a_yellow_wall2.png b/examples/a_plush_toy_robot_sitting_against_a_yellow_wall2.png
diff --git a/examples/a_shiba_inu_dog_wearing_a_beret_and_black_turtleneck1.png b/examples/a_shiba_inu_dog_wearing_a_beret_and_black_turtleneck1.png
diff --git a/examples/a_shiba_inu_dog_wearing_a_beret_and_black_turtleneck2.png b/examples/a_shiba_inu_dog_wearing_a_beret_and_black_turtleneck2.png
diff --git a/examples/an_abstract_painting_of_artificial_intelligence1.png b/examples/an_abstract_painting_of_artificial_intelligence1.png
diff --git a/examples/an_abstract_painting_of_artificial_intelligence2.png b/examples/an_abstract_painting_of_artificial_intelligence2.png
diff --git a/examples/an_armchair_in_the_shape_of_an_avocado.1.png b/examples/an_armchair_in_the_shape_of_an_avocado.1.png
diff --git a/examples/image1.png b/examples/image1.png
diff --git a/examples/image2.png b/examples/image2.png
diff --git a/src/imagen.py b/src/imagen.py
@@ -64,11 +64,11 @@ def upscale(self) -> None:
 
 if __name__ == "__main__":
     bearer = ""
-    prompt = "An oil painting of a Shiba Inu dog in a garden."
+    prompt = "3D render of a cute tropical fish in an aquarium on a dark blue background"
     demo = "BRAINSTORM"
     count = 2
     img = Imagen(bearer)
     images = img.generate(prompt, demo, count)
     for idx, image in enumerate(images):
         image.upscale()
-        image.save(f"image{idx+1}.png")
+        image.save(f"{prompt.replace(' ', '_').lower()}{idx + 1}.png")