Update app.py

app.py

@@ -6,6 +6,7 @@ from tensorflow.keras.preprocessing.image import load_img, img_to_array
 from tensorflow.keras.applications.densenet import preprocess_input, decode_predictions
 import numpy as np 
 from scipy import ndimage
+from skimage import exposure
 from PIL import Image
 import matplotlib.pyplot as plt
 import cv2
@@ -61,6 +62,7 @@ def custom_decode_predictions(predictions, class_labels):
 
 def classify_image(img):
     img_array = img_to_array(img)
+    img_array = exposure.equalize_hist(img_array)
     img_array = np.expand_dims(img_array, axis=0)
     img_array = preprocess_input(img_array)