MaxMilan1
commited on
Commit
·
0bbe8f6
1
Parent(s):
987f112
kjvb
Browse files
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 |
|