Updated images with pro prompts

.gitignore

@@ -133,6 +133,7 @@ src/doc2img/archive/
 
 # Model
 src/doc2img/best.pt
+src/try.ipynb
 
 # Experiment images
 #src/experiment_*

doc2img_data/stories/goldilocks.txt

@@ -0,0 +1,67 @@
+Once upon a time there were three Bears, who lived together in a house of their own, in a wood. One of them was a Little Wee Bear, and one was a Middle-sized Bear, and the other was a Great Big Bear. They had each a bowl for their porridge; a little bowl for the Little Wee Bear; and a middle-sized bowl for the Middle-sized Bear; and a great bowl for the Great Big Bear. And they had each a chair to sit in; a little chair for the Little Wee Bear; and a middle-sized chair for the Middle-sized Bear; and a great chair for the Great Big Bear. And they had each a bed to sleep in; a little bed for the Little Wee Bear; and a middle-sized bed for the Middle-sized Bear; and a great bed for the Great Big Bear.
+
+One day, after they had made the porridge for their breakfast, and poured it into their porridge-bowls, they walked out into the wood while the porridge was cooling, that they might not burn their mouths by beginning too soon, for they were polite, well-brought-up Bears. And while they were away a little girl called Goldilocks, who lived at the other side of the wood and had been sent on an errand by her mother, passed by the house, and looked in at the window. And then she peeped in at the keyhole, for she was not at all a well-brought-up little girl. Then seeing nobody in the house she lifted the latch. The door was not fastened, because the Bears were good Bears, who did nobody any harm, and never suspected that anybody would harm them. So Goldilocks opened the door and went in; and well pleased was she when she saw the porridge on the table. If she had been a well-brought-up little girl she would have waited till the Bears came home, and then, perhaps, they would have asked her to breakfast; for they were good Bears—a little rough or so, as the manner of Bears is, but for all that very good-natured and hospitable. But she was an impudent, rude little girl, and so she set about helping herself.
+
+First she tasted the porridge of the Great Big Bear, and that was too hot for her. Next she tasted the porridge of the Middle-sized Bear, but that was too cold for her. And then she went to the porridge of the Little Wee Bear, and tasted it, and that was neither too hot nor too cold, but just right, and she liked it so well that she ate it all up, every bit!
+
+Then Goldilocks, who was tired, for she had been catching butterflies instead of running on her errand, sate down in the chair of the Great Big Bear, but that was too hard for her. And then she sate down in the chair of the Middle-sized Bear, and that was too soft for her. But when she sat down in the chair of the Little Wee Bear, that was neither too hard nor too soft, but just right. So she seated herself in it, and there she sate till the bottom of the chair came out, and down she came, plump upon the ground; and that made her very cross, for she was a bad-tempered little girl.
+
+Now, being determined to rest, Goldilocks went upstairs into the bedchamber in which the Three Bears slept. And first she lay down upon the bed of the Great Big Bear, but that was too high at the head for her. And next she lay down upon the bed of the Middle-sized Bear, and that was too high at the foot for her. And then she lay down upon the bed of the Little Wee Bear, and that was neither too high at the head nor at the foot, but just right. So she covered herself up comfortably, and lay there till she fell fast asleep.
+
+By this time the Three Bears thought their porridge would be cool enough for them to eat it properly; so they came home to breakfast. Now careless Goldilocks had left the spoon of the Great Big Bear standing in his porridge.
+
+"SOMEBODY HAS BEEN AT MY PORRIDGE!"
+
+said the Great Big Bear in his great, rough, gruff voice.
+
+Then the Middle-sized Bear looked at his porridge and saw the spoon was standing in it too.
+
+"SOMEBODY HAS BEEN AT MY PORRIDGE!"
+
+said the Middle-sized Bear in his middle-sized voice.
+
+Then the Little Wee Bear looked at his, and there was the spoon in the porridge-bowl, but the porridge was all gone!
+
+"SOMEBODY HAS BEEN AT MY PORRIDGE, AND HAS EATEN IT ALL UP!"
+
+said the Little Wee Bear in his little wee voice.
+
+Upon this the Three Bears, seeing that some one had entered their house, and eaten up the Little Wee Bear's breakfast, began to look about them. Now the careless Goldilocks had not put the hard cushion straight when she rose from the chair of the Great Big Bear.
+
+"SOMEBODY HAS BEEN SITTING IN MY CHAIR!"
+
+said the Great Big Bear in his great, rough, gruff voice.
+
+And the careless Goldilocks had squatted down the soft cushion of the Middle-sized Bear.
+
+"SOMEBODY HAS BEEN SITTING IN MY CHAIR!"
+
+said the Middle-sized Bear in his middle-sized voice.
+
+"SOMEBODY HAS BEEN SITTING IN MY CHAIR, AND HAS SATE THE BOTTOM THROUGH!"
+
+said the Little Wee Bear in his little wee voice.
+
+Then the Three Bears thought they had better make further search in case it was a burglar, so they went upstairs into their bedchamber. Now Goldilocks had pulled the pillow of the Great Big Bear out of its place.
+
+"SOMEBODY HAS BEEN LYING IN MY BED!"
+
+said the Great Big Bear in his great, rough, gruff voice.
+
+And Goldilocks had pulled the bolster of the Middle-sized Bear out of its place.
+
+"SOMEBODY HAS BEEN LYING IN MY BED!"
+
+said the Middle-sized Bear in his middle-sized voice.
+
+But when the Little Wee Bear came to look at his bed, there was the bolster in its place! And the pillow was in its place upon the bolster!
+
+And upon the pillow——?
+
+There was Goldilocks's yellow head—which was not in its place, for she had no business there.
+
+"SOMEBODY HAS BEEN LYING IN MY BED,—AND HERE SHE IS STILL!"
+
+said the Little Wee Bear in his little wee voice.
+
+Now Goldilocks had heard in her sleep the great, rough, gruff voice of the Great Big Bear; but she was so fast asleep that it was no more to her than the roaring of wind, or the rumbling of thunder. And she had heard the middle-sized voice of the Middle-sized Bear, but it was only as if she had heard some one speaking in a dream. But when she heard the little wee voice of the Little Wee Bear, it was so sharp, and so shrill, that it awakened her at once. Up she started, and when she saw the Three Bears on one side of the bed, she tumbled herself out at the other, and ran to the window. Now the window was open, because the Bears, like good, tidy Bears, as they were, always opened their bedchamber window when they got up in the morning. So naughty, frightened little Goldilocks jumped; and whether she broke her neck in the fall, or ran into the wood and was lost there, or found her way out of the wood and got whipped for being a bad girl and playing truant, no one can say. But the Three Bears never saw anything more of her.

doc2img_data/stories/gpt.txt

@@ -0,0 +1,29 @@
+Once upon a time, in the vast African savannah, there lived a magnificent lion named Leo. He was known for his strength, grace, and golden mane that shimmered under the warm sun. Leo was a confident and fearless ruler of the animal kingdom.
+
+One day, as Leo was patrolling his territory, he noticed a small green frog hopping by his side. Surprised by this unusual encounter, Leo halted and looked down at the tiny amphibian. The frog, whose name was Freddie, looked up at the mighty lion and smiled.
+
+Leo's curiosity got the better of him. "What brings you here, little one?" he asked, his deep voice rumbling like distant thunder.
+
+Freddie replied, "I've always admired your bravery, Leo. I wanted to meet you and see if we could be friends."
+
+Leo was taken aback by Freddie's audacity but was also charmed by his courage. He decided to give the frog a chance and said, "Very well, Freddie. Let's embark on an adventure together."
+
+The unlikely duo set out on a journey beyond the savannah, venturing into new territories. They soon stumbled upon a towering snow-covered mountain. The sight was both mesmerizing and daunting. Leo had never encountered such cold and inhospitable terrain before.
+
+Freddie, however, remained undeterred. He had spent his life near lakes and ponds, adapting to various environments. With his webbed feet, he suggested, "Leo, I believe we can conquer this mountain together. I'll show you the way."
+
+The lion was hesitant at first, unsure if he was capable of enduring such challenging conditions. But seeing Freddie's determination and genuine friendship, Leo found his own courage reignited. "Lead the way, Freddie," he said with a newfound sense of purpose.
+
+As they climbed, the mountain's icy winds whipped against their faces, and the steep slopes tested their endurance. Leo struggled, slipping and sliding on the frozen ground. But Freddie encouraged him every step of the way, hopping ahead and leaving a trail in the snow for Leo to follow.
+
+Days turned into weeks, and their bond grew stronger with each passing obstacle. Leo learned to navigate the treacherous terrain, while Freddie found solace in the lion's unwavering support and protection.
+
+Finally, after an arduous climb, they reached the mountain's majestic peak. Leo's heart swelled with pride and triumph. The view from the summit was breathtaking—the world stretched out before them, painted in hues of blues, greens, and golds.
+
+As the snowflakes gently caressed their fur and skin, Leo turned to Freddie with gratitude shining in his eyes. "Thank you, my friend," he said, his voice filled with warmth. "I couldn't have done it without you."
+
+Freddie smiled, his little eyes sparkling. "And thank you, Leo, for believing in me and embracing the unexpected. Our friendship has made us stronger."
+
+From that day forward, Leo and Freddie became legends in the animal kingdom. Their tale of friendship and resilience spread far and wide, inspiring others to overcome their fears and conquer their own metaphorical mountains.
+
+And as long as the wind whistled through the snowy peaks, the story of Leo, the mighty lion, and Freddie, the brave little frog, continued to be shared, reminding everyone that true friendships can conquer any challenge, no matter how insurmountable.

doc2img_data/stories/hare_tortoise.txt

@@ -0,0 +1,10 @@
+A Hare was making fun of the Tortoise one day for being so slow.
+“Do you ever get anywhere?” he asked with a mocking laugh.
+“Yes,” replied the Tortoise, “and I get there sooner than you think. I’ll run you a race and prove it.”
+The Hare was much amused at the idea of running a race with the Tortoise, but for the fun of the thing he agreed. So the Fox, who had consented to act as judge, marked the distance and started the runners off.
+
+The Hare was soon far out of sight, and to make the Tortoise feel very deeply how ridiculous it was for him to try a race with a Hare, he lay down beside the course to take a nap until the Tortoise should catch up.
+
+The Tortoise meanwhile kept going slowly but steadily, and, after a time, passed the place where the Hare was sleeping. But the Hare slept on very peacefully; and when at last he did wake up, the Tortoise was near the goal. The Hare now ran his swiftest, but he could not overtake the Tortoise in time.
+
+Moral: The race is not always to the swift.

doc2img_data/stories/little_red_riding_hood.txt

@@ -0,0 +1,79 @@
+Once upon a time there was a dear little girl who was loved by every one who looked at her, but most of all by her grandmother, and there was nothing that she would not have given to the child. Once she gave her a little cap of red velvet, which suited her so well that she would never wear anything else. So she was always called Little Red Riding Hood.
+
+One day her mother said to her, "Come, Little Red Riding Hood, here is a piece of cake and a bottle of wine. Take them to your grandmother, she is ill and weak, and they will do her good. Set out before it gets hot, and when you are going, walk nicely and quietly and do not run off the path, or you may fall and break the bottle, and then your grandmother will get nothing. And when you go into her room, don't forget to say, good-morning, and don't peep into every corner before you do it."
+
+I will take great care, said Little Red Riding Hood to her mother, and gave her hand on it.
+
+The grandmother lived out in the wood, half a league from the village, and just as Little Red Riding Hood entered the wood, a wolf met her. Little Red Riding Hood did not know what a wicked creature he was, and was not at all afraid of him.
+
+"Good-day, Little Red Riding Hood," said he.
+
+"Thank you kindly, wolf."
+
+"Whither away so early, Little Red Riding Hood?"
+
+"To my grandmother's."
+
+"What have you got in your apron?"
+
+"Cake and wine. Yesterday was baking-day, so poor sick grandmother is to have something good, to make her stronger."
+
+"Where does your grandmother live, Little Red Riding Hood?"
+
+"A good quarter of a league farther on in the wood. Her house stands under the three large oak-trees, the nut-trees are just below. You surely must know it," replied Little Red Riding Hood.
+
+The wolf thought to himself, "What a tender young creature. What a nice plump mouthful, she will be better to eat than the old woman. I must act craftily, so as to catch both." So he walked for a short time by the side of Little Red Riding Hood, and then he said, "see Little Red Riding Hood, how pretty the flowers are about here. Why do you not look round. I believe, too, that you do not hear how sweetly the little birds are singing. You walk gravely along as if you were going to school, while everything else out here in the wood is merry."
+
+Little Red Riding Hood raised her eyes, and when she saw the sunbeams dancing here and there through the trees, and pretty flowers growing everywhere, she thought, suppose I take grandmother a fresh nosegay. That would please her too. It is so early in the day that I shall still get there in good time. And so she ran from the path into the wood to look for flowers. And whenever she had picked one, she fancied that she saw a still prettier one farther on, and ran after it, and so got deeper and deeper into the wood.
+
+Meanwhile the wolf ran straight to the grandmother's house and knocked at the door.
+
+"Who is there?"
+
+"Little Red Riding Hood," replied the wolf. "She is bringing cake and wine. Open the door."
+
+"Lift the latch," called out the grandmother, "I am too weak, and cannot get up."
+
+The wolf lifted the latch, the door sprang open, and without saying a word he went straight to the grandmother's bed, and devoured her. Then he put on her clothes, dressed himself in her cap, laid himself in bed and drew the curtains.
+
+Little Red Riding Hood, however, had been running about picking flowers, and when she had gathered so many that she could carry no more, she remembered her grandmother, and set out on the way to her.
+
+She was surprised to find the cottage-door standing open, and when she went into the room, she had such a strange feeling that she said to herself, oh dear, how uneasy I feel to-day, and at other times I like being with grandmother so much.
+
+She called out, "Good morning," but received no answer. So she went to the bed and drew back the curtains. There lay her grandmother with her cap pulled far over her face, and looking very strange.
+
+"Oh, grandmother," she said, "what big ears you have."
+
+"The better to hear you with, my child," was the reply.
+
+"But, grandmother, what big eyes you have," she said.
+
+"The better to see you with, my dear."
+
+"But, grandmother, what large hands you have."
+
+"The better to hug you with."
+
+"Oh, but, grandmother, what a terrible big mouth you have."
+
+"The better to eat you with."
+
+And scarcely had the wolf said this, than with one bound he was out of bed and swallowed up Little Red Riding Hood.
+
+When the wolf had appeased his appetite, he lay down again in the bed, fell asleep and began to snore very loud. The huntsman was just passing the house, and thought to himself, how the old woman is snoring. I must just see if she wants anything.
+
+So he went into the room, and when he came to the bed, he saw that the wolf was lying in it. "Do I find you here, you old sinner," said he. "I have long sought you."
+
+Then just as he was going to fire at him, it occurred to him that the wolf might have devoured the grandmother, and that she might still be saved, so he did not fire, but took a pair of scissors, and began to cut open the stomach of the sleeping wolf.
+
+When he had made two snips, he saw the Little Red Riding Hood shining, and then he made two snips more, and the little girl sprang out, crying, "Ah, how frightened I have been. How dark it was inside the wolf."
+
+And after that the aged grandmother came out alive also, but scarcely able to breathe. Little Red Riding Hood, however, quickly fetched great stones with which they filled the wolf's belly, and when he awoke, he wanted to run away, but the stones were so heavy that he collapsed at once, and fell dead.
+
+Then all three were delighted. The huntsman drew off the wolf's skin and went home with it. The grandmother ate the cake and drank the wine which Little Red Riding Hood had brought, and revived, but Little Red Riding Hood thought to herself, as long as I live, I will never by myself leave the path, to run into the wood, when my mother has forbidden me to do so.
+
+It is also related that once when Little Red Riding Hood was again taking cakes to the old grandmother, another wolf spoke to her, and tried to entice her from the path. Little Red Riding Hood, however, was on her guard, and went straight forward on her way, and told her grandmother that she had met the wolf, and that he had said good-morning to her, but with such a wicked look in his eyes, that if they had not been on the public road she was certain he would have eaten her up. "Well," said the grandmother, "we will shut the door, that he may not come in."
+
+Soon afterwards the wolf knocked, and cried, "open the door, grandmother, I am Little Red Riding Hood, and am bringing you some cakes."
+
+But they did not speak, or open the door, so the grey-beard stole twice or thrice round the house, and at last jumped on the roof, intending to wait until Little Red Riding Hood went home in the evening, and then to steal after her and devour her in the darkness. But the grandmother saw what was in his thoughts. In front of the house was a great stone trough, so she said to the child, take the pail, Little Red Riding Hood. I made some sausages yesterday, so carry the water in which I boiled them to the trough. Little Red Riding Hood carried until the great trough was quite full. Then the smell of the sausages reached the wolf, and he sniffed and peeped down, and at last stretched out his neck so far that he could no longer keep his footing and began to slip, and slipped down from the roof straight into the great trough, and was drowned. But Little Red Riding Hood went joyously home, and no one ever did anything to harm her again.