mbar0075 commited on
Commit
8365aad
·
1 Parent(s): 25f8e99

Fixed minor changes

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -56,8 +56,8 @@ def predict_image(image, model_name: str, size=(244, 244)) -> List[Tuple[str, st
56
  maltese_categories = categories[model_name]["maltese"]
57
  sorted_predictions = [
58
  (
59
- english_categories[i],
60
- maltese_categories[i],
61
  round(pred_probs[i] * 100, 2) # Convert to percentage
62
  )
63
  for i in sorted_indices
 
56
  maltese_categories = categories[model_name]["maltese"]
57
  sorted_predictions = [
58
  (
59
+ english_categories[str(i)],
60
+ maltese_categories[str(i)],
61
  round(pred_probs[i] * 100, 2) # Convert to percentage
62
  )
63
  for i in sorted_indices