|
import gradio as gr |
|
from datasets import load_dataset |
|
from PIL import Image |
|
|
|
import re |
|
import os |
|
import requests |
|
|
|
from share_btn import community_icon_html, loading_icon_html, share_js |
|
|
|
word_list_dataset = load_dataset("stabilityai/word-list", data_files="list.txt", use_auth_token=True) |
|
word_list = word_list_dataset["train"]['text'] |
|
|
|
is_gpu_busy = False |
|
def infer(person1, person2, scale): |
|
global is_gpu_busy |
|
|
|
prompt = f"Cross between {person1} and {person2}" |
|
for filter in word_list: |
|
if re.search(rf"\b{filter}\b", prompt): |
|
raise gr.Error("Unsafe content found. Please try again with different prompts.") |
|
|
|
images = [] |
|
person_1 = '' |
|
url = os.getenv('JAX_BACKEND_URL') |
|
payload = {'prompt': prompt, 'guidance_scale': scale} |
|
images_request = requests.post(url, json = payload) |
|
for image in images_request.json()["images"]: |
|
image_b64 = (f"data:image/jpeg;base64,{image}") |
|
images.append(image_b64) |
|
|
|
return images |
|
|
|
|
|
css = """ |
|
.gradio-container { |
|
font-family: 'IBM Plex Sans', sans-serif; |
|
} |
|
.gr-button { |
|
color: white; |
|
border-color: black; |
|
background: black; |
|
} |
|
input[type='range'] { |
|
accent-color: black; |
|
} |
|
.dark input[type='range'] { |
|
accent-color: #dfdfdf; |
|
} |
|
.container { |
|
max-width: 730px; |
|
margin: auto; |
|
padding-top: 1.5rem; |
|
} |
|
#gallery { |
|
min-height: 22rem; |
|
margin-bottom: 15px; |
|
margin-left: auto; |
|
margin-right: auto; |
|
border-bottom-right-radius: .5rem !important; |
|
border-bottom-left-radius: .5rem !important; |
|
} |
|
#gallery>div>.h-full { |
|
min-height: 20rem; |
|
} |
|
.details:hover { |
|
text-decoration: underline; |
|
} |
|
.gr-button { |
|
white-space: nowrap; |
|
} |
|
.gr-button:focus { |
|
border-color: rgb(147 197 253 / var(--tw-border-opacity)); |
|
outline: none; |
|
box-shadow: var(--tw-ring-offset-shadow), var(--tw-ring-shadow), var(--tw-shadow, 0 0 #0000); |
|
--tw-border-opacity: 1; |
|
--tw-ring-offset-shadow: var(--tw-ring-inset) 0 0 0 var(--tw-ring-offset-width) var(--tw-ring-offset-color); |
|
--tw-ring-shadow: var(--tw-ring-inset) 0 0 0 calc(3px var(--tw-ring-offset-width)) var(--tw-ring-color); |
|
--tw-ring-color: rgb(191 219 254 / var(--tw-ring-opacity)); |
|
--tw-ring-opacity: .5; |
|
} |
|
#advanced-btn { |
|
font-size: .7rem !important; |
|
line-height: 19px; |
|
margin-top: 12px; |
|
margin-bottom: 12px; |
|
padding: 2px 8px; |
|
border-radius: 14px !important; |
|
} |
|
#advanced-options { |
|
display: none; |
|
margin-bottom: 20px; |
|
} |
|
.footer { |
|
margin-bottom: 45px; |
|
margin-top: 35px; |
|
text-align: center; |
|
border-bottom: 1px solid #e5e5e5; |
|
} |
|
.footer>p { |
|
font-size: .8rem; |
|
display: inline-block; |
|
padding: 0 10px; |
|
transform: translateY(10px); |
|
background: white; |
|
} |
|
.dark .footer { |
|
border-color: #303030; |
|
} |
|
.dark .footer>p { |
|
background: #0b0f19; |
|
} |
|
.acknowledgments h4{ |
|
margin: 1.25em 0 .25em 0; |
|
font-weight: bold; |
|
font-size: 115%; |
|
} |
|
.animate-spin { |
|
animation: spin 1s linear infinite; |
|
} |
|
@keyframes spin { |
|
from { |
|
transform: rotate(0deg); |
|
} |
|
to { |
|
transform: rotate(360deg); |
|
} |
|
} |
|
#share-btn-container { |
|
display: flex; padding-left: 0.5rem !important; padding-right: 0.5rem !important; background-color: #000000; justify-content: center; align-items: center; border-radius: 9999px !important; width: 13rem; |
|
margin-top: 10px; |
|
margin-left: auto; |
|
} |
|
#share-btn { |
|
all: initial; color: #ffffff;font-weight: 600; cursor:pointer; font-family: 'IBM Plex Sans', sans-serif; margin-left: 0.5rem !important; padding-top: 0.25rem !important; padding-bottom: 0.25rem !important;right:0; |
|
} |
|
#share-btn * { |
|
all: unset; |
|
} |
|
#share-btn-container div:nth-child(-n+2){ |
|
width: auto !important; |
|
min-height: 0px !important; |
|
} |
|
#share-btn-container .wrap { |
|
display: none !important; |
|
} |
|
|
|
.gr-form{ |
|
flex: 1 1 50%; border-top-right-radius: 0; border-bottom-right-radius: 0; |
|
} |
|
#prompt-container{ |
|
gap: 0; |
|
} |
|
#prompt-text-input, #negative-prompt-text-input{padding: .45rem 0.625rem} |
|
#component-16{border-top-width: 1px!important;margin-top: 1em} |
|
.image_duplication{position: absolute; width: 100px; left: 50px} |
|
""" |
|
|
|
block = gr.Blocks(css=css) |
|
|
|
examples = [ |
|
[ |
|
'Gollum', |
|
'Legolas', |
|
9 |
|
], |
|
[ |
|
'Frodo', |
|
'Gandalf', |
|
9 |
|
], |
|
[ |
|
'Gollum', |
|
'Captain Picard', |
|
9 |
|
], |
|
[ |
|
'Harry Potter', |
|
'Dobby', |
|
9 |
|
], |
|
[ |
|
"Yoda", |
|
'a Borg drone', |
|
9 |
|
], |
|
] |
|
|
|
|
|
with block: |
|
gr.HTML( |
|
""" |
|
<div style="text-align: center; margin: 0 auto;"> |
|
<div |
|
style=" |
|
display: inline-flex; |
|
align-items: center; |
|
gap: 0.8rem; |
|
font-size: 1.75rem; |
|
" |
|
> |
|
<svg |
|
width="0.65em" |
|
height="0.65em" |
|
viewBox="0 0 115 115" |
|
fill="none" |
|
xmlns="http://www.w3.org/2000/svg" |
|
> |
|
<rect width="23" height="23" fill="white"></rect> |
|
<rect y="69" width="23" height="23" fill="white"></rect> |
|
<rect x="23" width="23" height="23" fill="#AEAEAE"></rect> |
|
<rect x="23" y="69" width="23" height="23" fill="#AEAEAE"></rect> |
|
<rect x="46" width="23" height="23" fill="white"></rect> |
|
<rect x="46" y="69" width="23" height="23" fill="white"></rect> |
|
<rect x="69" width="23" height="23" fill="black"></rect> |
|
<rect x="69" y="69" width="23" height="23" fill="black"></rect> |
|
<rect x="92" width="23" height="23" fill="#D9D9D9"></rect> |
|
<rect x="92" y="69" width="23" height="23" fill="#AEAEAE"></rect> |
|
<rect x="115" y="46" width="23" height="23" fill="white"></rect> |
|
<rect x="115" y="115" width="23" height="23" fill="white"></rect> |
|
<rect x="115" y="69" width="23" height="23" fill="#D9D9D9"></rect> |
|
<rect x="92" y="46" width="23" height="23" fill="#AEAEAE"></rect> |
|
<rect x="92" y="115" width="23" height="23" fill="#AEAEAE"></rect> |
|
<rect x="92" y="69" width="23" height="23" fill="white"></rect> |
|
<rect x="69" y="46" width="23" height="23" fill="white"></rect> |
|
<rect x="69" y="115" width="23" height="23" fill="white"></rect> |
|
<rect x="69" y="69" width="23" height="23" fill="#D9D9D9"></rect> |
|
<rect x="46" y="46" width="23" height="23" fill="black"></rect> |
|
<rect x="46" y="115" width="23" height="23" fill="black"></rect> |
|
<rect x="46" y="69" width="23" height="23" fill="black"></rect> |
|
<rect x="23" y="46" width="23" height="23" fill="#D9D9D9"></rect> |
|
<rect x="23" y="115" width="23" height="23" fill="#AEAEAE"></rect> |
|
<rect x="23" y="69" width="23" height="23" fill="black"></rect> |
|
</svg> |
|
<h1 style="font-weight: 900; margin-bottom: 7px;margin-top:5px"> |
|
GANDalf 0.0 Demo |
|
</h1> |
|
</div> |
|
<p style="margin-bottom: 10px; font-size: 94%; line-height: 23px;"> |
|
One Shot Face Tolkeinization with Diffusers. |
|
</p> |
|
</div> |
|
""" |
|
) |
|
with gr.Group(): |
|
with gr.Box(): |
|
with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True): |
|
with gr.Column(): |
|
person1 = gr.Textbox( |
|
label="First Person", |
|
show_label=False, |
|
max_lines=1, |
|
placeholder="First person", |
|
elem_id="person1-text-input", |
|
).style( |
|
border=(True, False, True, True), |
|
rounded=(True, False, False, True), |
|
container=False, |
|
) |
|
person2 = gr.Textbox( |
|
label="Second Person", |
|
show_label=False, |
|
max_lines=1, |
|
placeholder="Second person", |
|
elem_id="person2-text-input", |
|
).style( |
|
border=(True, False, True, True), |
|
rounded=(True, False, False, True), |
|
container=False, |
|
) |
|
btn = gr.Button("Generate image").style( |
|
margin=False, |
|
rounded=(False, True, True, False), |
|
full_width=False, |
|
) |
|
|
|
gallery = gr.Gallery( |
|
label="Generated images", show_label=False, elem_id="gallery" |
|
).style(grid=[2], height="auto") |
|
|
|
with gr.Group(elem_id="container-advanced-btns"): |
|
|
|
with gr.Group(elem_id="share-btn-container"): |
|
community_icon = gr.HTML(community_icon_html) |
|
loading_icon = gr.HTML(loading_icon_html) |
|
share_button = gr.Button("Share to community", elem_id="share-btn") |
|
|
|
with gr.Accordion("Advanced settings", open=False): |
|
|
|
|
|
|
|
guidance_scale = gr.Slider( |
|
label="Guidance Scale", minimum=0, maximum=50, value=9, step=0.1 |
|
) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
ex = gr.Examples(examples=examples, fn=infer, inputs=[person1, person2, guidance_scale], outputs=[gallery, community_icon, loading_icon, share_button], cache_examples=False) |
|
ex.dataset.headers = [""] |
|
negative.submit(infer, inputs=[person1, person2, guidance_scale], outputs=[gallery], postprocess=False) |
|
text.submit(infer, inputs=[person1, person2, guidance_scale], outputs=[gallery], postprocess=False) |
|
btn.click(infer, inputs=[person1, person2, guidance_scale], outputs=[gallery], postprocess=False) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
share_button.click( |
|
None, |
|
[], |
|
[], |
|
_js=share_js, |
|
) |
|
gr.HTML( |
|
""" |
|
<div class="footer"> |
|
<p>Model by <a href="https://huggingface.co/hlydecker" style="text-decoration: underline;" target="_blank">hydecker</a> - backend running for free in this Gradio Demo by 🤗 Hugging Face |
|
</p> |
|
</div> |
|
""" |
|
) |
|
with gr.Accordion(label="License", open=False): |
|
gr.HTML( |
|
"""<div class="acknowledgments"> |
|
<p><h4>LICENSE</h4> |
|
GANDalf is basically just a prompt wrapper for meme purposes on top of Stable Diffusion. |
|
The model is licensed with a <a href="https://huggingface.co/stabilityai/stable-diffusion-2/blob/main/LICENSE-MODEL" style="text-decoration: underline;" target="_blank">CreativeML OpenRAIL++</a> license. The authors claim no rights on the outputs you generate, you are free to use them and are accountable for their use which must not go against the provisions set in this license. The license forbids you from sharing any content that violates any laws, produce any harm to a person, disseminate any personal information that would be meant for harm, spread misinformation and target vulnerable groups. For the full list of restrictions please <a href="https://huggingface.co/spaces/CompVis/stable-diffusion-license" target="_blank" style="text-decoration: underline;" target="_blank">read the license</a></p> |
|
<p><h4>Biases and content acknowledgment</h4> |
|
Despite how impressive being able to turn text into image is, beware to the fact that this model may output content that reinforces or exacerbates societal biases, as well as realistic faces, pornography and violence. The model was trained on the <a href="https://laion.ai/blog/laion-5b/" style="text-decoration: underline;" target="_blank">LAION-5B dataset</a>, which scraped non-curated image-text-pairs from the internet (the exception being the removal of illegal content) and is meant for research purposes. You can read more in the <a href="https://huggingface.co/CompVis/stable-diffusion-v1-4" style="text-decoration: underline;" target="_blank">model card</a></p> |
|
</div> |
|
""" |
|
) |
|
|
|
|
|
block.queue(concurrency_count=80, max_size=100).launch(max_threads=150) |