Add application file

app.py

@@ -51,6 +51,7 @@ with open("scene_labels.json") as labels_file:
 
 def classify_image(img):
     img = tf.convert_to_tensor(img, dtype=tf.float32)
+    img = tf.expand_dims(img, axis=0)
     # pred,idx,probs = model.predict(img)
     probs = model.predict(img)
     return {labels[i]: float(probs[i]) for i in range(len(labels))}