MaxMilan1 commited on
Commit
0bbe8f6
·
1 Parent(s): 987f112
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -23,6 +23,7 @@ pipe.to("cuda")
23
  def generate_image(prompt, neg_prompt):
24
  prompt += "no background, side view, minimalist shot"
25
  image_bytes = pipe(prompt, negative_prompt=neg_prompt)
 
26
  pil_image = Image.open(BytesIO(image_bytes))
27
  opencvImage = cv2.cvtColor(numpy.array(pil_image), cv2.COLOR_RGB2BGR)
28
 
 
23
  def generate_image(prompt, neg_prompt):
24
  prompt += "no background, side view, minimalist shot"
25
  image_bytes = pipe(prompt, negative_prompt=neg_prompt)
26
+ print(image_bytes)
27
  pil_image = Image.open(BytesIO(image_bytes))
28
  opencvImage = cv2.cvtColor(numpy.array(pil_image), cv2.COLOR_RGB2BGR)
29