Update README.md

README.md

@@ -6,19 +6,19 @@ tags:
 - diffusers
 - fal
 widget:
-  - text: "an intricately designed perfume bottle resting on a vintage carved dressing table, flanked by elegant makeup and jewelry, with a gilded mirror above and a vase of peonies adding a romantic ambiance, all depicted in a refined vintage style."
+  - text: "Fiery red and orange lettering against a dark charcoal background, with the letters appearing to be made of flickering flames and glowing embers, giving a sense of intense heat and dynamic movement. The texture should mimic the crackling and flowing nature of fire, with occasional sparks flying off the edges."
     output:
       url: pictures/t2i_1.png
-  - text: "A lively image featuring a bottle of alcopop on a sleek bar counter, a bokeh effect of an enthusiastic crowd and a live band with colorful stage lighting in the background, all wrapped up in a vivid urban nightlife ambiance"
+  - text: "Cool blue and turquoise lettering against a deep navy background, with the letters appearing to be made of flowing water and gentle waves, giving a sense of fluidity and calm. The texture should mimic the rippling and shimmering surface of a clear ocean, with light reflections and occasional droplets splashing off the edges."
     output:
       url: pictures/t2i_2.png
-  - text: "A luxurious crossbody bag is the centerpiece on a glass shelf, with a selection of high-end accessories neatly arrayed behind it, all bathed in a soft, diffused sunlight that adds a serene and exquisite touch to the image."
+  - text: "Creamy pastel-colored lettering against a light, frosty background, with the letters appearing to be made of swirled, soft-serve ice cream, giving a sense of deliciousness and indulgence. The texture should mimic the smooth, velvety surface of freshly scooped ice cream, with subtle swirls, drips, and a slightly glossy, mouth-watering finish."
     output:
       url: pictures/t2i_3.png
-  - text: "The table is placed in an outdoor camping scene with a teapot and two teacups on the table, grass, outdoor, sunlight, HD, product photography."
+  - text: "Vibrant, multicolored lettering against a soft, pastel background, with the letters appearing to be made of delicate petals and blooming flowers, giving a sense of freshness and natural beauty. The texture should mimic the intricate layers and velvety surfaces of various blossoms, with subtle gradients and occasional dewdrops enhancing the lifelike appearance."
     output:
       url: pictures/t2i_4.png
-  - text: "A bottle is placed in water, the water covers half of the product, the splash flies, the bottom half of the product placed under the water is clearly visible, waterproof advertising, product advertising, photography, HD, real, 8k"
+  - text: "Rich, bold lettering against a textured canvas background, with the letters appearing to be made of thick, vibrant oil paint strokes, giving a sense of depth and artistic expression. The texture should mimic the dynamic, layered application of oil paints, with visible brushstrokes, impasto effects, and a glossy finish that catches the light in different ways."
     output:
       url: pictures/t2i_5.png
   - text: "A photo of a serene landscape with a calm lake reflecting the surrounding mountains and forests. The mountains have a mix of tree and exposed rock. The sky is overcast, with dark clouds covering the sky. The reflection of the mountains and sky on the lake is mirror-clear. The shoreline is lined with trees. The ground is covered with grass."
@@ -51,11 +51,8 @@ pipe = FluxControlNetPipeline.from_pretrained('black-forest-labs/FLUX.1-dev',
                                               torch_dtype=torch.bfloat16).to("cuda")
 
 
-font_mask_pil = Image.open(font_mask_path)
+font_mask_pil = Image.open(font_mask_path).convert("RGB")
 font_mask_npy = np.array(font_mask_pil)
-if len(font_mask_npy.shape) == 2:
-   font_mask_npy = cv2.cvtColor(font_mask_npy, cv2.COLOR_GRAY2RGB)
-   font_mask_pil = Image.fromarray(font_mask_npy)
 
 prompt = "Vibrant, multicolored lettering against a soft, pastel background, with the letters appearing to be made of delicate petals and blooming flowers, giving a sense of freshness and natural beauty. The texture should mimic the intricate layers and velvety surfaces of various blossoms, with subtle gradients and occasional dewdrops enhancing the lifelike appearance."
 image = pipe(prompt,
@@ -64,7 +61,8 @@ image = pipe(prompt,
              num_inference_steps=30,
              guidance_scale=3.5,
              generator=torch.Generator("cuda").manual_seed(42)).images[0]
-image.save("./result.png")
+rgba = Image.fromarray(np.concatenate([np.array(image), cv2.resize(font_mask_npy, (1024, 1024))[..., :1]], axis=-1))
+rgba.save("./{}.png".format(datetime.now().strftime("%Y%m%d%H%M%S")))
 ```