Upload 3 files

app.py

@@ -0,0 +1,50 @@
+import streamlit as st
+import random
+import re
+from transformers import pipeline, set_seed
+
+gpt2_pipe = pipeline('text-generation', model='Gustavosta/MagicPrompt-Stable-Diffusion', tokenizer='gpt2')
+
+with open("examples.txt", "r") as f:
+    line = f.readlines()
+
+def generate(starting_text):
+    seed = random.randint(100, 1000000)
+    set_seed(seed)
+
+    if starting_text == "":
+        starting_text: str = line[random.randrange(0, len(line))].replace("\n", "").lower().capitalize()
+        starting_text: str = re.sub(r"[,:\-–.!;?_]", '', starting_text)
+
+    response = gpt2_pipe(starting_text, max_length=(len(starting_text) + random.randint(60, 90)), num_return_sequences=4)
+    response_list = []
+    for x in response:
+        resp = x['generated_text'].strip()
+        if resp != starting_text and len(resp) > (len(starting_text) + 4) and resp.endswith((":", "-", "—")) is False:
+            response_list.append(resp+'\n')
+
+    response_end = "\n".join(response_list)
+    response_end = re.sub('[^ ]+\.[^ ]+','', response_end)
+    response_end = response_end.replace("<", "").replace(">", "")
+
+    if response_end != "":
+        return response_end
+
+st.title("Stable Diffusion Prompt Creator")
+
+st.markdown("This is a web app for [this](https://huggingface.co/Gustavosta/MagicPrompt-Stable-Diffusion) model trained by Gustavosta for Stable Diffusion to create a Prompt from a few words. You can submit your own text or select from provided examples.")
+
+starting_text = st.text_input(label="Initial Text", placeholder="Text here", value="")
+
+if st.button("Generate"):
+    result = generate(starting_text)
+    st.write("<div style='background-color: #2E2E2E; padding: 10px;'>{}</div>".format("<br>".join(result.splitlines())), unsafe_allow_html=True)
+
+examples = []
+for x in range(5):
+    examples.append(line[random.randrange(0, len(line))].replace("\n", "").lower().capitalize())
+
+st.write("")
+st.write("<div style='text-align: center; font-weight: bold;'>Examples:</div>",unsafe_allow_html=True)
+for example in examples:
+    st.write("<div style='background-color: #2E2E2E; padding: 10px; text-align: center;'>• {}</div>".format(example), unsafe_allow_html=True)

examples.txt

@@ -0,0 +1,139 @@
+The Dusk of the Damned Hamlet
+The Blackwell Brigade Circus.
+The Pilgrim of the Paradox
+The Blackened Blight of Vengeance
+The Depths of the Cosmic Calamity
+The Treasonous Treacherous Throne
+The Bloodthirsty Bandit's Booty
+The Enchanted Estate of the Epic Endgame
+The Enchanted Enclave of the End
+A Precious Prize from Frankenstein's Lab
+The Priestess and the Prophecy
+The Hideaway that Prague Hid
+Among the Branches and the Blackbirds
+The Enchanted Enclave of the Eclipse
+The Laughter of the Living Dead's Last Laugh
+Among the Branches and the Belligerent Blackbirds
+The Treacherous Trickster Throne
+Into the Depths of the Cosmic Horror
+The Priceless Prize from Frankenstein's Frightening Factory
+The Priestess and the Ravens
+The Pilgrim of the Past
+The Depths of the Cosmic Catastrophe
+The Mechanics of the Matrix
+The Jailer's Chilly Chains
+The Emergence of the Enchanting Enchantress
+The Brutal Bounty of the Bandit
+Among the Boughs and the Beautiful Blackbirds
+The Province of the Prophet
+The Flames of the Fateful Studio
+The Goblins of the Glitzy Glee and Glamorous Glitz
+The Dark Dominion of the Dictators
+The Lucky Losses of the Gambler
+The Enchanted Enclave of the Endgame
+The Secret Treacherous Throne
+The Majestic Maleficent Monarchs
+The Priestess and the Pyrotechnics
+The Silent Shadow Overlords
+The Rise of the She-Wolf
+The Priestess and the Pyre
+The Precious Prize from Frankenstein's Foundry
+The Treacherous Trickster's Tainted Throne
+The Mechanics of the Maze
+The Hideout that Prague Hid in the Hinterlands
+The Mechanisms of the Mind
+The Treasonous Trickster's Triumphant Throne
+The Majestic Malevolent Monarchies of Misery
+The Machines of the Malevolent Matrix
+The Blackwell Bro Circus Brigade
+The Goblins of the Glamour and Glitz
+The Quest for the Queen
+The Lodge that Prague Left Behind
+The Blackwell Bro Circus Band
+The Treasonous Trickster’s Throne
+The Jailer's Chilly Clutches
+The Precious Prize from Frankenstein's Forge
+The Enchanted Expedition
+The Legacy of the Legendary Legends
+The Jailer’s Chilly Clutch
+The Cryptic Crowned Commanders
+The Grandiose Grim Monarchs
+The Fortunate Fortune of the Fortuneteller's Fate
+The Mechanics of the Merciless Maze
+The Fortunate Fate of the Fortune Teller
+The Fortunate Fate of the Fortuneteller
+The Sunset of the Sullen Hamlet
+To Laugh or to Die
+The Hideout that Prague Hid Away
+The Blackwell Bro Circus.
+The Quest for the Queen of the Quicksilver
+The Province of the Powerful Prophet
+The Quest for the Queen of the Quagmire
+The Secret Sovereigns of the Shadowlands of Sorrow
+The Brutal Bounty of the Bandit King
+The Emperor in the Freezer of Winter
+The Goblins of the Glee and Glitzy Glamour
+The Secret Sovereigns of the Shadowlands
+The Harmony of the Hallowed Studio
+The Depths of the Cosmic Catastrophic Chaos
+The Jailer's Cold Clasp
+The Goblins of the Glitz and Glamour
+The Sunset of the Sinister Hamlet
+The Sunset of the Sorrowful Hamlet
+The Dark Domain of the Dictatorship
+The Jailer's Chill
+The Emperor in the Frigid Frost of Winter
+The Mysterious Dungeon of Despair
+The Property of the Pope
+The Legacy of the Legendaries
+The Goblins of the Glee and Glitz
+The Depths of the Cosmic Cataclysm
+The Hideaway that Prague Held Hostage
+The Emperor in the Flames of Winter
+The Bloodthirsty Bandit's Bounty
+The Awakening of the Amazonian Warrior
+The Goblins of the Glitzy Glee and Glittering Glamour
+The Twilight of the Tragic Hamlet
+The Laughter of the Living Dead
+The Twilight of the Tragic Tragedy of Hamlet
+Among the Roots and the Ravens
+The Steam-Powered Stallion of the Stunning Spectacle
+The Steam-Powered Strongman of the Stage
+The Priceless Prize from Frankenstein’s Factory
+The Blackwell Brothers Circus.
+The Legacy of the Legendary Heroes
+The Majestic malevolent Monarchs
+The Precious Prize from Frankenstein's Frightening Furnace
+The Legacy of the Legend
+The Priestess and the Pyromania
+The Emperor in the Frigid Freezer of the Frozen Winter
+Among the Branches and the Batty Birds
+The Emperor in the Frost of Winter
+The Laugh of the Living Dead
+The Pilgrim of the Perplexing Paradox
+The Champions of the Champion
+The Pilgrim of the Puzzling Paradox
+The Pilgrim of the Present
+The Quest for the Queen of the Quicksand
+The Majestic Malevolent Monarchs
+The Harmony of the Hallowed House of Horror Studio
+The Machines of the Mind
+The Awakening of the Amazon
+The Fortunate Fortune of the Gambler
+The Depths of the Cosmic Horror
+The Grandiose Grim Monarchies
+The Emergence of the Enchantress
+The Enchanted Estate of the Endgame
+The Emperor in the Frigid Freezer of Winter
+The Steam-Powered Showman of Strength
+Among the Boughs and the Blackcaps
+The Province of the Prince
+The Bloodthirsty Bandit's Bountiful Booty
+The Harmony of the Haunted House Studio
+The Harmony of the Haunted Studio
+The Steam-Powered Stallion of the Stage
+The Dominion of the Daring Duke
+The Hideout that Prague Hid in the Hills
+The Laughter of the Lurking Dead
+The Dominion of the Duke
+The Steam-Powered Stallion of the Spectacle

requirements.txt

@@ -0,0 +1,4 @@
+jinja2==3.0.3
+transformers==4.22.2
+sentencepiece
+torch