change to jpg

app.py

@@ -9,6 +9,7 @@ import segmentation_models as sm
 from keras.metrics import MeanIoU
 from functools import partial
 from glob import glob
+from PIL import Image
 
 
 
@@ -57,7 +58,8 @@ model.compile(metrics=metrics)
 '#images/human_pixel_masks.npy'
 def predict(ash_image, model=model):
     #label = np.load(label_image)
-    ash_image = np.load(ash_image)[...,4]
+    #ash_image = np.load(ash_image)[...,4]
+    ash_image = np.asarray(Image.open(ash_image))
     y_pred = model.predict(ash_image.reshape(1,256, 256, 3))
     prediction = np.argmax(y_pred[0], axis=2).reshape(256,256,1)
     #intersection = label & prediction
@@ -81,7 +83,7 @@ if __name__ == "__main__":
         section_btn = gr.Button("Generate Predictions")
         section_btn.click(partial(predict, model=model), img_input, img_output)
 
-        images_dir = glob(os.path.join(os.getcwd(), "images") + os.sep + "*.png")
+        images_dir = glob(os.path.join(os.getcwd(), "images") + os.sep + "*.jpg")
         examples = [i for i in images_dir]
         gr.Examples(examples=examples, inputs=img_input, outputs=img_output)