Harveenchadha commited on
Commit
ba795af
·
1 Parent(s): 1b6ea60

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -10,10 +10,12 @@ model = from_pretrained_keras("Harveenchadha/low-light-image-enhancement", compi
10
  #examples = ['examples/179.png', 'examples/493.png', 'examples/780.png']
11
 
12
 
13
- def infer(original_image):
 
14
  image = keras.preprocessing.image.img_to_array(original_image)
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
  # print([len(a) for a in output])
 
10
  #examples = ['examples/179.png', 'examples/493.png', 'examples/780.png']
11
 
12
 
13
+ def infer(original_image):
14
+ print(original_image.shape())
15
  image = keras.preprocessing.image.img_to_array(original_image)
16
  image = image.astype("float32") / 255.0
17
  image = np.expand_dims(image, axis=0)
18
+ print(image.shape())
19
  output = model.predict(image)
20
  # print(len(output))
21
  # print([len(a) for a in output])