Fixed class labels

app.py

@@ -5,8 +5,8 @@ from huggingface_hub import from_pretrained_fastai
 learn_imp = from_pretrained_fastai('MikeTrizna/Prunus_lineage_classifier')
 
 def predict(image_np, description=None):
-  classes = ['neotropical racemose', 'paleotropical racemose',
-             'temperate diploid', 'temperate racemose']
+  classes = ['Neotropical racemose', 'Paleotropical racemose',
+             'Solitary/Corymbose', 'Temperate racemose']
   pred,pred_idx,probs = learn_imp.predict(image_np)
   confidences = {classes[idx]: f'{prob:.04f}' for idx, prob in enumerate(probs)}
   return confidences