xinlongwang commited on
Commit
4519f9b
·
1 Parent(s): 4c82dea
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -22,7 +22,10 @@ def inference_mask1(prompt,
22
  a = json.loads(r.text)
23
  res = []
24
  for i in range(len(a)):
25
- res.append(np.uint8(np.array(Image.open(io.BytesIO(base64.b64decode(a[i]))))))
 
 
 
26
  return res
27
 
28
  def resizeImg(img):
 
22
  a = json.loads(r.text)
23
  res = []
24
  for i in range(len(a)):
25
+ out = Image.open(io.BytesIO(base64.b64decode(a[i])))
26
+ out = out.resize((224, 224))
27
+ res.append(np.uint8(np.array(out)))
28
+ #res.append(np.uint8(np.array(Image.open(io.BytesIO(base64.b64decode(a[i]))))))
29
  return res
30
 
31
  def resizeImg(img):