Update README.md

README.md

@@ -43,6 +43,8 @@ More details [here](https://github.com/lucasdegeorge/T2I-ImageNet/blob/main/READ
 
 ### CAD-I model
 
+In this repo, the model is trained with **Text Augmentation and Image Augmentation**. Check the model trained with Text Augmentation only [here](https://huggingface.co/Lucasdegeorge/CAD-I_TA)
+
 To use the pre-trained model do the following:
 ```python
 from pipe import T2IPipeline
@@ -70,17 +72,13 @@ Our models have been specifically trained to handle very long and detailed promp
 Example prompts: 
 
 ```
-A vintage Polaroid camera sits inside a crinkled plastic bag, partially buried in the damp sand of a quiet beach. The transparent plastic clings to the camera, slightly fogged with condensation, distorting the faded colors of the once-pristine device. Gentle waves roll in the background, their foamy edges reaching toward the shore, while the golden hues of the setting sun cast long shadows across the sand. Small flecks of salt and tiny grains of sand stick to the bag, hinting at the passage of time and exposure to the elements.",
-A raccoon stands upright in a meticulously designed astronaut suit, its ringed tail swaying gracefully beneath the reinforced fabric in defiance of zero gravity. The suit is a masterwork of engineering—pristine white panels seamlessly integrated with brushed titanium accents and carbon-fiber joints, while iridescent mission patches catch starlight on its shoulders. Through the helmet's gold-tinted visor, keen eyes shine with a mix of mischief and wonder, and the reflective surface mirrors an endless ballet of celestial bodies: pinprick stars scattered like diamond dust, the marble-like swirls of a distant gas giant, and ribbons of nebulae painting the void in ethereal purples and blues. A life-support backpack hugs its frame like a technological shell, status indicators pulsing with soft blue light, while around the suited explorer, motes of cosmic dust drift like suspended glitter, catching the light of a distant sun.","A quirky crab, entirely sculpted from various types of yellow cheese, sits proudly on a white plate. Its body is a smooth, golden wheel of cheese, round and rich in color, while soft, creamy cheese legs extend outward in neatly shaped segments, each one gently curled as if the crab is about to scuttle away. The claws are crafted from sharp, crumbly yellow cheese, carefully carved to resemble the pincers, with tiny bits of grated Parmesan scattered across to give it texture. The eyes are tiny olives, carefully set in place with delicate toothpicks, adding a playful touch to the cheese creation. Surrounding the crab, fresh basil leaves are placed to resemble seaweed, completing the amusing and mouthwatering oceanic scene on the plate.",
-A golden retriever, its fur glowing like molten gold, sits serenely at the rim of a volcano, seemingly unbothered by the intense heat and chaos around it. The dog's soft, flowing coat contrasts sharply with the fiery landscape as it gazes calmly into the fiery abyss. Lava bubbles and pours from the crater, casting an eerie, amber glow that illuminates the dog's fur in a warm, almost ethereal light. The air shimmers with heat, while plumes of dark smoke rise into the sky, swirling like ominous clouds above. Despite the tumultuous eruption, the retriever exudes a quiet confidence, its relaxed posture and wagging tail creating an almost surreal sense of calm within the violent eruption. It's as if this brave dog has claimed the heart of the volcano as its own kingdom, a noble ruler amidst the storm of fire and ash.",
-An adorable otter, with its sleek, brown fur and bright, curious eyes, playfully interacts with a vibrant bunch of broccoli. The otter, balanced on its back, holds the broccoli in its small paws, gently nibbling on a floret with a look of contentment. The broccoli's deep green florets stand out against the otter's playful, natural hues, creating a fun contrast between the two. The scene unfolds by the edge of a peaceful riverbank, where the otter has momentarily paused its swimming adventure to enjoy a healthy snack. The soft ripples of the water reflect the sunlight, adding to the serene, lighthearted moment of the otter and its green, crunchy companion.",
-A wooden ostrich stands quietly in the center of an empty museum hall, its smooth, dark timber body gleaming under the soft, ambient light that filters through tall, arched windows. The ostrich's long, slender neck curves gracefully, its polished surface reflecting the muted tones of the space, from deep mahogany browns to hints of golden amber. The intricate grain of the wood adds texture and depth to its form, giving it a life-like quality despite its static position. Around the body, layers of carved wood feathers mimic the natural texture of plumage, adding elegance and a sculptural dimension. The bird's long, powerful legs are crafted with care, positioned in a gentle, poised stance. Its head, elongated and smooth, slopes into a sharply defined beak, with small, round insets of dark wood forming the eyes, giving the sculpture an expressive, almost lifelike gaze. The surrounding museum space is vast and quiet, with empty walls and clean, polished floors, which only amplify the serene presence of the wooden ostrich standing as a timeless piece of art in the middle of the room. The stillness of the museum creates a peaceful contrast to the life captured in the graceful figure of the ostrich, as the wooden sculpture seems to watch over the space, adding a touch of nature to the otherwise silent, empty environment.",
-A bold, electric green owl with oversized sunglasses lounges on a desk in the center of a bright, modern classroom. Its vivid green feathers pop against the soft, neutral tones of the room, from the dark wood of the desks to the pale blue walls adorned with educational posters. The owl's sharp eyes peek over the edge of its sleek sunglasses, exuding an aura of cool confidence, like it's ready to drop some serious wisdom. Sunlight pours through the classroom windows, making its colorful feathers shimmer and creating a playful contrast against the seriousness of the chalkboard and neatly arranged desks. The owl's wings are slightly raised, as if caught mid-gesture, adding to the sense of relaxed energy it brings to the space. This odd combination of the owl's laid-back vibe and the structured classroom setting creates an amusing, surreal scene—one that feels both out of place and effortlessly cool.",
-A curious fox sits comfortably in front of a glowing screen, its fluffy tail curled neatly around its paws. The soft light from the screen casts a gentle glow on the fox's bright orange fur, highlighting its pointed ears and sharp eyes that are locked intently on the display. Its fur contrasts beautifully against the dark room, where shadows stretch across the floor and walls. The fox's gaze is focused, almost as if it's trying to make sense of what's happening on the screen, its whiskers twitching with curiosity. The surrounding atmosphere is quiet, save for the soft hum of the device in front of it, adding a sense of peaceful solitude. The entire scene feels whimsical, with the fox's natural grace and the modern technology blending in an unexpectedly charming way.",
-The image portrays the iconic Iron Throne from *Game of Thrones*, but reimagined entirely out of jagged, crystalline ice, radiating a cold, otherworldly aura. The throne's structure is massive and imposing, with hundreds of sharp, glistening shards of ice jutting out at chaotic angles, resembling frozen blades. The pale blue hue of the ice contrasts starkly against the dark, shadowy hall surrounding it, illuminated only by faint, icy light filtering through narrow windows. Frost creeps across the stone floor, and thin tendrils of mist swirl around the base of the throne, adding an eerie, mystical atmosphere. The seat itself is smooth but treacherous, with a faint glow emanating from within the ice, as if it holds some ancient, frozen power. The scene evokes a sense of both awe and dread, as though the throne itself is alive, a symbol of dominance forged in the cold, unyielding grip of winter.",
-The image centers on a powerful lion, its golden fur gleaming under a soft, ethereal light as it skillfully controls a bright orange basketball. The lion's massive paws handle the ball with surprising dexterity, its sharp claws carefully tucked away to avoid damaging it. Its muscular body moves with fluid grace, every motion exuding strength and precision as it dribbles with focused intensity. The lion's piercing eyes lock onto the ball, its expression a mix of determination and wild instinct, while its majestic mane flows dynamically with each movement. The basketball, slightly frosted as if touched by the cool night air, contrasts vividly against the lion's natural, earthy tones. The scene captures a striking blend of raw animal power and unexpected playfulness, creating a surreal and captivating moment.",
-The image showcases a strikingly elegant black cat, its fur an inky, absolute black so deep and rich that it seems to swallow all light, creating a velvety void of darkness. Its eyes, a vivid, electric green, blaze with intensity, glowing like neon against the cat's shadowy form, creating a dramatic, high-contrast visual. The green is so vibrant it almost pulses, drawing immediate attention and giving the cat an otherworldly, hypnotic presence. The cat sits perfectly still, its posture regal and composed, with every sleek line of its body accentuated by the stark interplay of light and shadow. The background is muted and dark, almost indistinguishable, ensuring the cat's jet-black fur and luminous eyes dominate the scene entirely. This bold contrast between the absolute black and the radiant green makes the image mesmerizing, almost surreal, and impossible to look away from.",
-
+A majestic elephant stands tall and proud in the heart of the African savannah, its wrinkled, gray skin glistening under the intense afternoon sun. The elephant's large, flapping ears and long, sweeping trunk create a sense of grace and power as it gently sways, surveying the vast, golden grasslands stretching out before it. In the distance, a herd of zebras grazes peacefully, their stripes blending with the tall, dry grass. The scene is completed by a lone acacia tree silhouetted against the setting sun, casting long, dramatic shadows across the landscape.
+A classic film camera rests on a tripod, its worn leather strap and scratched metal body telling the story of countless adventures and captured moments. The camera is positioned in a scenic landscape, with rolling hills, a winding river, and a distant mountain range bathed in the soft, golden light of sunset. In the foreground, a wildflower meadow sways gently in the breeze, while the camera's lens captures the beauty and tranquility of the scene, preserving it for eternity.
+A graceful flamingo stands elegantly in the shallow waters of a tranquil lagoon, its vibrant pink feathers reflecting beautifully in the still water. The flamingo's long, slender legs and curved neck create a picture of poise and balance as it dips its beak into the water, searching for food. Behind the flamingo, a lush mangrove forest stretches out, its dense foliage providing a rich habitat for various wildlife. The scene is completed by a clear blue sky and the gentle rustling of leaves in the breeze
+A hearty, overstuffed sandwich sits on a wooden cutting board, its layers of fresh, crisp lettuce, juicy tomatoes, and thinly sliced deli meats peeking out from between two slices of golden-brown bread. The sandwich's tantalizing aroma fills the air, mingling with the scent of freshly baked bread and tangy mustard. In the background, a bustling deli comes to life, with shelves lined with jars of pickles, a gleaming meat slicer, and a chalkboard menu listing the day's specials. The scene is completed by the lively chatter of customers and the clinking of glasses.
+A stunning oil painting of a majestic tiger hangs on the wall of a dimly-lit art gallery, its vibrant colors and intricate details drawing the viewer in. The tiger's powerful, muscular body is depicted in mid-stride, its stripes blending seamlessly with the lush jungle foliage surrounding it. The painting captures the tiger's intense, amber eyes and the subtle play of light and shadow on its fur, creating a sense of depth and movement. The background features a dense canopy of trees and a cascading waterfall, adding to the wild, untamed atmosphere of the scene.
+A clever magpie perched on a rustic wooden fence post, its iridescent black and white feathers shimmering in the sunlight. The bird tilts its head, holding a shiny trinket in its beak, with a backdrop of a golden wheat field swaying gently in the breeze. A few more curios and found objects are scattered along the fence, hinting at the magpie's treasure trove hidden nearby. A clear blue sky with puffy white clouds completes the scenic countryside atmosphere.
+A playful dolphin leaps gracefully out of the sparkling turquoise waters, its sleek, gray body arcing through the air before diving back into the waves with a splash. Nearby, a classic wooden sailboat glides smoothly across the ocean, its white sails billowing in the breeze. The dolphin swims alongside the boat, its joyful antics mirrored by the shimmering sunlight dancing on the water's surface. The scene is completed by a clear blue sky and the distant horizon, where the sea meets the sky
 ```