Spaces:
Runtime error
Runtime error
File size: 2,620 Bytes
937f590 36fb421 93dd707 937f590 93dd707 cd7f3ee 93dd707 cd7f3ee 829fa44 937f590 c91c54c 4f8e7f5 c91c54c 51088c7 c91c54c b88a1b2 7713986 937f590 5c0e5df 937f590 d5b9aaa b991d0a 4f8e7f5 04448f6 937f590 4f8e7f5 c91c54c d7040df c91c54c 36fb421 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 |
# !pip install -Uq diffusers transformers
# !pip install -Uq gradio
# !pip install -Uq accelerate
import gradio
from diffusers import StableDiffusionPipeline as pipeline
from accelerate import init_empty_weights
import torch
import os
api_key = os.environ['api_key']
my_token = api_key
with init_empty_weights():
pipe = pipeline.from_pretrained("CompVis/stable-diffusion-v1-4", revision="fp16", torch_dtype=torch.float16, use_auth_token=my_token).to("cuda")
DROPDOWNS = {
"gustav": " by dan mumford and gustav klimt and john harris and jean delville and victo ngai and josan gonzalez",
"hayao": " by studio ghibli",
"vinny": " painting by Vincent van Gogh",
"danny": " drawn by a child",
}
def image_prompt(prompt, dropdown):
prompt = prompt + DROPDOWNS[dropdown]
return pipe(prompt=prompt, height=512, width=512).images[0]
with gradio.Blocks(css="""
#go-button {
background-color: white;
border-radius: 0;
border: none;
font-family: serif;
background-image: none;
font-weight: 100;
width: fit-content;
display: block;
margin-left: auto;
margin-right: auto;
text-decoration: underline;
box-shadow: none;
color: blue;
}
.rounded-lg {
border: none;
}
.gr-box {
border-radius: 0;
border: 1px solid black;
}
.text-gray-500 {
color: black;
font-family: serif;
font-size: 15px;
}
.border-gray-200 {
border: 1px solid black;
}
.bg-gray-200 {
background-color: white;
--tw-bg-opacity: 0;
}
footer {
display: none;
}
footer {
opacity: 0;
}
#output-image {
border: 1px solid black;
background-color: white;
width: 500px;
display: block;
margin-left: auto;
margin-right: auto;
}
.absolute {
display: none;
}
#input-text {
width: 500px;
display: block;
margin-left: auto;
margin-right: auto;
padding: 0 0 0 0;
}
.py-6 {
padding-top: 0;
padding-bottom: 0;
}
.px-4 {
padding-left: 0;
padding-right: 0;
}
.rounded-lg {
border-radius: 0;
}
.gr-padded {
padding: 0 0;
margin-bottom: 12.5px;
}
.col > *, .col > .gr-form > * {
width: 500px;
margin-left: auto;
margin-right: auto;
}
""") as demo:
dropdown = gradio.Dropdown(["danny", "gustav", "hayao", "vinny"], label="choose style...")
prompt = gradio.Textbox(label="image prompt...", elem_id="input-text")
output = gradio.Image(elem_id="output-image")
go_button = gradio.Button("draw it!", elem_id="go-button")
go_button.click(fn=image_prompt, inputs=[prompt, dropdown], outputs=output)
demo.launch() |