Harveenchadha commited on
Commit
fd6857d
Β·
1 Parent(s): 7bca426

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -17,7 +17,9 @@ def infer(original_image):
17
  output = model.predict(image)
18
  output_image = output[0] * 255.0
19
  output_image = output_image.clip(0, 255)
20
- print(output_image.shape())
 
 
21
  output_image = output_image.reshape(
22
  (np.shape(output_image)[0], np.shape(output_image)[1], 3)
23
  )
 
17
  output = model.predict(image)
18
  output_image = output[0] * 255.0
19
  output_image = output_image.clip(0, 255)
20
+ print(output_image)
21
+ print(len(output_image))
22
+ print([len(a) for a in output_image])
23
  output_image = output_image.reshape(
24
  (np.shape(output_image)[0], np.shape(output_image)[1], 3)
25
  )