Spaces:
Runtime error
Runtime error
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 |
|