Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -18,7 +18,7 @@ controlnet = ControlNetModel.from_pretrained(
|
|
| 18 |
)
|
| 19 |
|
| 20 |
pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
|
| 21 |
-
"
|
| 22 |
controlnet=controlnet,
|
| 23 |
safety_checker=None,
|
| 24 |
torch_dtype=torch.float16,
|
|
@@ -34,9 +34,9 @@ def inference(
|
|
| 34 |
input_image: Image.Image,
|
| 35 |
prompt: str,
|
| 36 |
negative_prompt: str,
|
| 37 |
-
guidance_scale: float =
|
| 38 |
-
controlnet_conditioning_scale: float = 1
|
| 39 |
-
strength: float = 0.
|
| 40 |
seed: int = -1,
|
| 41 |
sampler = "DPM++ Karras SDE",
|
| 42 |
):
|
|
|
|
| 18 |
)
|
| 19 |
|
| 20 |
pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
|
| 21 |
+
"SG161222/Realistic_Vision_V2.0",
|
| 22 |
controlnet=controlnet,
|
| 23 |
safety_checker=None,
|
| 24 |
torch_dtype=torch.float16,
|
|
|
|
| 34 |
input_image: Image.Image,
|
| 35 |
prompt: str,
|
| 36 |
negative_prompt: str,
|
| 37 |
+
guidance_scale: float = 8.0,
|
| 38 |
+
controlnet_conditioning_scale: float = 1,
|
| 39 |
+
strength: float = 0.9,
|
| 40 |
seed: int = -1,
|
| 41 |
sampler = "DPM++ Karras SDE",
|
| 42 |
):
|