nielsgl commited on
Commit
21b1026
1 Parent(s): 562d4fa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -14,7 +14,7 @@ resolution = 512
14
  # checkpoint of the converted Stable Diffusion from KerasCV
15
  model_ckpt = "nielsgl/dreambooth-bored-ape"
16
  pipeline = StableDiffusionPipeline.from_pretrained(model_ckpt)
17
- # pipeline.to("cuda")
18
 
19
  unique_id = "drawbayc"
20
  class_label = "monkey"
@@ -41,9 +41,9 @@ output = gr.Gallery(label="Outputs").style(grid=(1,2))
41
  title = "KerasCV Stable Diffusion Demo on images of Bored Apes."
42
  description = "This is a dreambooth model fine-tuned on images the NFT collection of the Bored Ape Yacht Club. To try it, input the concept with `drawbayc ape`."
43
  examples=[
44
- ["A drawing of a drawbayc ape dressed as a cowboy", "bad anatomy, blurry, ugly, deformed, disfigured", 12, 50],
45
- ["A drawing of a drawbayc ape dressed as a clown", "bad anatomy, blurry, ugly, deformed, disfigured", 12, 50],
46
- ["A drawing of a drawbayc ape dressed as a turtle", "bad anatomy, blurry, ugly, deformed, disfigured", 12, 50],
47
 
48
  ]
49
 
 
14
  # checkpoint of the converted Stable Diffusion from KerasCV
15
  model_ckpt = "nielsgl/dreambooth-bored-ape"
16
  pipeline = StableDiffusionPipeline.from_pretrained(model_ckpt)
17
+ pipeline = pipeline.to("cuda")
18
 
19
  unique_id = "drawbayc"
20
  class_label = "monkey"
 
41
  title = "KerasCV Stable Diffusion Demo on images of Bored Apes."
42
  description = "This is a dreambooth model fine-tuned on images the NFT collection of the Bored Ape Yacht Club. To try it, input the concept with `drawbayc ape`."
43
  examples=[
44
+ ["A drawing of a drawbayc ape dressed as a cowboy", "bad anatomy, blurry, ugly, deformed, disfigured", 12, 75],
45
+ ["A drawing of a drawbayc ape dressed as a clown", "bad anatomy, blurry, ugly, deformed, disfigured", 12, 75],
46
+ ["A drawing of a drawbayc ape dressed as a turtle", "bad anatomy, blurry, ugly, deformed, disfigured", 12, 75],
47
 
48
  ]
49