Update app.py

app.py

@@ -11,20 +11,17 @@ model = from_pretrained_keras("Harveenchadha/low-light-image-enhancement", compi
 
 
 def infer(original_image):
-    
     image = keras.preprocessing.image.img_to_array(original_image)
     image = image.astype("float32") / 255.0
     image = np.expand_dims(image, axis=0)
-    
     output = model.predict(image)
-    # print(len(output))
-    # print([len(a) for a in output])
     output_image = output[0] * 255.0
     output_image = output_image.clip(0, 255)
     output_image = output_image.reshape(
-        (np.shape(image)[0], np.shape(image)[1], 3)
+        (np.shape(output_image)[0], np.shape(output_image)[1], 3)
     )
     output_image = np.uint32(output_image)
+    return output_image
     
     # output_image = tf.cast((output[0, :, :, :] * 255), dtype=np.uint8)