MaxMilan1
commited on
Commit
·
c0a0649
1
Parent(s):
d8aa11d
change
Browse files- util/text_img.py +1 -4
util/text_img.py
CHANGED
@@ -47,10 +47,7 @@ def generate_image(prompt, control_image, controlnet):
|
|
47 |
elif controlnet == "Depth":
|
48 |
control_image = get_depth(control_image)
|
49 |
|
50 |
-
image = pipe(prompt,
|
51 |
-
negative_prompt=negative_prompt,
|
52 |
-
image=control_image,
|
53 |
-
controlnet_conditioning_scale=1.0).images[0]
|
54 |
|
55 |
image2 = rembg.remove(image)
|
56 |
|
|
|
47 |
elif controlnet == "Depth":
|
48 |
control_image = get_depth(control_image)
|
49 |
|
50 |
+
image = pipe(prompt, image=control_image).images[0]
|
|
|
|
|
|
|
51 |
|
52 |
image2 = rembg.remove(image)
|
53 |
|