Fixed minor changes
Browse files
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
|