Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -20,7 +20,7 @@ from diffusers import DiffusionPipeline, StableDiffusionPipeline, StableDiffusio
|
|
| 20 |
#from diffusers import DiffusionPipeline
|
| 21 |
#pipe = DiffusionPipeline.from_pretrained("mswhite/nnn-model")
|
| 22 |
|
| 23 |
-
pipe = StableDiffusionPipeline.from_pretrained("mswhite/
|
| 24 |
pipe = pipe.to("cuda")
|
| 25 |
pipe.safety_checker = lambda images, **kwargs: (images, [False] * len(images))
|
| 26 |
|
|
@@ -64,10 +64,10 @@ def nagelize(text):
|
|
| 64 |
|
| 65 |
with gr.Blocks() as demo:
|
| 66 |
|
| 67 |
-
gr.Markdown("<img src=https://huggingface.co/spaces/mswhite/
|
| 68 |
output = gr.Image(label="Output").style(width=720)
|
| 69 |
art_to_draw = gr.Textbox(label="Prompt to Draw: e.g. ")
|
| 70 |
-
flashb_btn = gr.Button("
|
| 71 |
flashb_btn.click(fn=nagelize, inputs=[art_to_draw], outputs=[output])
|
| 72 |
|
| 73 |
demo.launch()
|
|
|
|
| 20 |
#from diffusers import DiffusionPipeline
|
| 21 |
#pipe = DiffusionPipeline.from_pretrained("mswhite/nnn-model")
|
| 22 |
|
| 23 |
+
pipe = StableDiffusionPipeline.from_pretrained("mswhite/bricksburg_model", torch_dtype=torch.float16) #, safety_checker=None)
|
| 24 |
pipe = pipe.to("cuda")
|
| 25 |
pipe.safety_checker = lambda images, **kwargs: (images, [False] * len(images))
|
| 26 |
|
|
|
|
| 64 |
|
| 65 |
with gr.Blocks() as demo:
|
| 66 |
|
| 67 |
+
gr.Markdown("<img src=https://huggingface.co/spaces/mswhite/Bricksburg/resolve/main/Emmet.jpg width=720px>")
|
| 68 |
output = gr.Image(label="Output").style(width=720)
|
| 69 |
art_to_draw = gr.Textbox(label="Prompt to Draw: e.g. ")
|
| 70 |
+
flashb_btn = gr.Button("Everything is Awesome in Bricksburg")
|
| 71 |
flashb_btn.click(fn=nagelize, inputs=[art_to_draw], outputs=[output])
|
| 72 |
|
| 73 |
demo.launch()
|