Update app.py

app.py

@@ -15,8 +15,9 @@ def image_mod(image):
     resize = tf.image.resize(img, (256, 256))
     plt.imshow(resize.numpy().astype(int))
     yhat = model.predict(np.expand_dims(resize,0))
-    display = np.argmax(yhat)
-    display = str(yhat)
+    display_old = np.argmax(yhat)
+    display = str(display_old)
+    #display = str(yhat)
     #display = str(display)
     if display == "0":
         message = "Cloudy" #