Update app.py

app.py

@@ -47,7 +47,6 @@ id2label = config["id2label"]
 label2id = config["label2id"]
 image_size = config["image_size"]
 classes_names = list(label2id.keys())
-print(id2label)
 '''
 # import labels
 classes_names = ["Acropore_branched", "Acropore_digitised", "Acropore_tabular", "Algae_assembly", 
@@ -77,7 +76,7 @@ def predict(input_image):
     result = {}
     i = 0
     for score in scores:
-        label = id2label[i]
+        label = classes_names[i]
         result[label] = float(score)
         i += 1
     result = {key: result[key] for key in result if result[key] > 0.5}