datt30 commited on
Commit
9c0e205
·
1 Parent(s): 5311371

added image to test behavior

Browse files
Files changed (1) hide show
  1. 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
- control_image,
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,