Update app.py
Browse files
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 |
-
|
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,
|
45 |
-
["A drawing of a drawbayc ape dressed as a clown", "bad anatomy, blurry, ugly, deformed, disfigured", 12,
|
46 |
-
["A drawing of a drawbayc ape dressed as a turtle", "bad anatomy, blurry, ugly, deformed, disfigured", 12,
|
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 |
|