adpro commited on
Commit
f31255e
·
1 Parent(s): 88f9ae9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -41,6 +41,8 @@ def img2img_generate(source_img, prompt, steps=25, strength=0.75, seed=42, guida
41
  print("prompt: ", prompt)
42
  buffered = BytesIO()
43
  source_img.save(buffered, format="JPEG")
 
 
44
  img_b64 = base64.b64encode(buffered.getvalue())
45
 
46
  data = {"source_img": img_b64.decode(), "prompt": prompt, "steps": steps,
 
41
  print("prompt: ", prompt)
42
  buffered = BytesIO()
43
  source_img.save(buffered, format="JPEG")
44
+ simg = cv2.imread(source_img)
45
+ print(simg.size)
46
  img_b64 = base64.b64encode(buffered.getvalue())
47
 
48
  data = {"source_img": img_b64.decode(), "prompt": prompt, "steps": steps,