Harveenchadha commited on
Commit
cef63f3
Β·
1 Parent(s): 87da6f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ def infer(original_image):
15
  image = image.astype("float32") / 255.0
16
  image = np.expand_dims(image, axis=0)
17
  output = model.predict(image)
18
- print(output[0].shape(), output[1].shape())
19
  '''output_image = output[0] * 255.0
20
  output_image = output_image.clip(0, 255)
21
  output_image = output_image.reshape(
 
15
  image = image.astype("float32") / 255.0
16
  image = np.expand_dims(image, axis=0)
17
  output = model.predict(image)
18
+ print(len(output))
19
  '''output_image = output[0] * 255.0
20
  output_image = output_image.clip(0, 255)
21
  output_image = output_image.reshape(