Update app.py

app.py

@@ -86,8 +86,11 @@ def make_prediction(image, taxonomic_resolution_choice, taxonomic_level):
             # Move to the next higher taxonomic level
             current_taxonomic_level_index += 1
 
-        # Update the taxonomic level for output
-        taxonomic_level = taxonomic_levels[current_taxonomic_level_index]
+        # Ensure we don't go out of bounds
+        if current_taxonomic_level_index < len(taxonomic_levels):
+            taxonomic_level = taxonomic_levels[current_taxonomic_level_index]
+        else:
+            taxonomic_level = taxonomic_levels[-1]  # fallback to the highest taxonomic level
     else:
         # Aggregate predictions based on the selected taxonomic level
         aggregated_predictions, aggregated_class_labels = aggregate_predictions(prediction, taxonomic_level, class_names)