Spaces:
Sleeping
Sleeping
datt30
commited on
Commit
·
9c0e205
1
Parent(s):
5311371
added image to test behavior
Browse files- geo_painting.py +5 -1
geo_painting.py
CHANGED
@@ -4,6 +4,7 @@ from diffusers import (
|
|
4 |
UniPCMultistepScheduler,
|
5 |
ControlNetModel
|
6 |
)
|
|
|
7 |
|
8 |
|
9 |
class GeoPainting:
|
@@ -29,9 +30,12 @@ class GeoPainting:
|
|
29 |
def generate_painting(self, input_promp, control_image):
|
30 |
print(":::::::::::::::::")
|
31 |
print(type(control_image))
|
|
|
|
|
|
|
32 |
output = self.pipe(
|
33 |
input_promp,
|
34 |
-
|
35 |
negative_prompt="ugly, disfigured, low quality, blurry, nsfw",
|
36 |
generator=self.generator,
|
37 |
num_inference_steps=20,
|
|
|
4 |
UniPCMultistepScheduler,
|
5 |
ControlNetModel
|
6 |
)
|
7 |
+
from diffusers.utils import load_image
|
8 |
|
9 |
|
10 |
class GeoPainting:
|
|
|
30 |
def generate_painting(self, input_promp, control_image):
|
31 |
print(":::::::::::::::::")
|
32 |
print(type(control_image))
|
33 |
+
|
34 |
+
image = load_image("sparkgeo_logo.png")
|
35 |
+
|
36 |
output = self.pipe(
|
37 |
input_promp,
|
38 |
+
image,
|
39 |
negative_prompt="ugly, disfigured, low quality, blurry, nsfw",
|
40 |
generator=self.generator,
|
41 |
num_inference_steps=20,
|