Update app.py

app.py

@@ -15,6 +15,8 @@ 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))
+    if yhat == "":
+        yhat = "None"
     #display = np.argmax(yhat)
     #display = str(display)
     #if display == "0":