Update app.py

app.py

@@ -18,9 +18,9 @@ def classify_image(inp):
   inp = inp.reshape((-1, HEIGHT,WIDTH, 3))
   inp = tf.keras.applications.nasnet.preprocess_input(inp) 
   prediction = model.predict(inp)
-  labels = dict((v,k) for k,v in labels.items())
+  label = dict((v,k) for k,v in labels.items())
   predicted_class_indices=np.argmax(prediction,axis=1)
-  return {labels[i]: float(predicted_class_indices[i]) for i in range(NUM_CLASSES)}
+  return {label[i]: float(predicted_class_indices[i]) for i in range(NUM_CLASSES)}