Update app.py

app.py

@@ -61,13 +61,13 @@ def predict(ash_image, model=model):
     #label = np.load(label_image)
     # ash_image = np.load('images/552609781892851211.npy')
     print(images_means)
-
-    im_mean = round(np.asarray(ash_image).mean(),4)
+    im = np.asarray(Image.open(ash_image))
+    im_mean = round(im.mean(),4)
     print(im_mean)
     if im_mean in images_means:
-        ash_image = np.load('images/' + images_means[im_mean])
+        im = np.load('images/' + images_means[im_mean])
 
-    y_pred = model.predict(np.asarray(ash_image).reshape(1,256, 256, 3))
+    y_pred = model.predict(im.reshape(1,256, 256, 3))
     prediction = np.argmax(y_pred[0], axis=2).reshape(256,256)
     #intersection = label & prediction
     #false_negative = label - intersection
@@ -83,7 +83,7 @@ if __name__ == "__main__":
     with gr.Blocks(title="Contrail Predictions") as demo:
         gr.Markdown("""<h1><center>Predict Contrails in Satellite Images</center></h1>""")
         with gr.Row():
-            img_input = gr.Image(type="pil", height=256, width=256, label="Image Input")
+            img_input = gr.Image(type="filepath", height=256, width=256, label="Image Input")
             img_output = gr.AnnotatedImage(label="Predictions", height=256, width=256, color_map=class2hexcolor)
 
         section_btn = gr.Button("Generate Predictions")