Update app.py

app.py

@@ -21,7 +21,12 @@ def classify_image(inp):
   prediction = model.predict(inp)
   #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)}
+  result = {}
+  for i in range(len(predicted_class_indices)):
+      if predicted_class_indices[i] < NUM_CLASSES:
+          result[labels[predicted_class_indices[i]]]= float(predicted_class_indices[i])
+  return result 
+