Update app.py
Browse filestry to get rid of error
app.py
CHANGED
|
@@ -26,13 +26,13 @@ control_img = 'myimage.jpg'
|
|
| 26 |
|
| 27 |
examples = [["a yellow dog in grass", "lowres, two heads, bad muzzle, bad anatomy, missing ears, missing paws", "example1.jpg"]]
|
| 28 |
|
| 29 |
-
default_example = examples[0]
|
| 30 |
|
| 31 |
controlnet, controlnet_params = FlaxControlNetModel.from_pretrained(
|
| 32 |
"JFoz/dog-cat-pose", dtype=jnp.bfloat16
|
| 33 |
)
|
| 34 |
pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
|
| 35 |
-
"runwayml/stable-diffusion-v1-5", controlnet=controlnet, revision="flax", dtype=jnp.bfloat16
|
| 36 |
)
|
| 37 |
|
| 38 |
def infer(prompts, negative_prompts, image):
|
|
|
|
| 26 |
|
| 27 |
examples = [["a yellow dog in grass", "lowres, two heads, bad muzzle, bad anatomy, missing ears, missing paws", "example1.jpg"]]
|
| 28 |
|
| 29 |
+
#default_example = examples[0]
|
| 30 |
|
| 31 |
controlnet, controlnet_params = FlaxControlNetModel.from_pretrained(
|
| 32 |
"JFoz/dog-cat-pose", dtype=jnp.bfloat16
|
| 33 |
)
|
| 34 |
pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
|
| 35 |
+
"runwayml/stable-diffusion-v1-5", controlnet=controlnet, revision="flax", dtype=jnp.bfloat16, safety_checker=None,
|
| 36 |
)
|
| 37 |
|
| 38 |
def infer(prompts, negative_prompts, image):
|