Update app.py

app.py

@@ -16,6 +16,7 @@ if file is not None:
     x  = tf.image.resize(x, [80, 80])
     x = np.repeat(x[:, :, np.newaxis], 3, axis=2)
     x = np.squeeze(x)
+    x  = np.expand_dims(x, axis=0)
     prediction = model.predict(x)
     print(prediction)