mswhite commited on
Commit
09344e2
·
verified ·
1 Parent(s): e555ce7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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/nnn-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,10 +64,10 @@ def nagelize(text):
64
 
65
  with gr.Blocks() as demo:
66
 
67
- gr.Markdown("<img src=https://huggingface.co/spaces/mswhite/NNN/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("Go to Legoland")
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()