Update app.py

Remove bugs from custome_image_processing function

app.py

@@ -7,15 +7,13 @@ import gradio as gr
 from keras.models import load_model
 
 def custom_Image_preprocessing(image_data, target_size=(150, 150)):
-    img = Image.fromarray((image_data * 255).astype('uint8'))
-    img = img.resize(target_size)
-    img_arr = np.array(img)
-    if img_arr.shape[-1] != 3:
-        img_arr = np.stack((img_arr,) * 3, axis=-1)
+    img = image.array_to_img(image_data, data_format='channels_last')
+    img = img.resize(target_size)  # Resize the image if needed
+    img_arr = image.img_to_array(img)
+    img_arr = img_arr * 1./255
     img_arr = np.expand_dims(img_arr, axis=0)
     return img_arr
 
-
 # function to predict the custome image
 
 def image_predict(image_path):