Update app.py

app.py

@@ -60,7 +60,13 @@ def make_prediction(image, taxonomic_decision, taxonomic_level):
     
     # Initialize variables for aggregated predictions and level index
     aggregated_predictions = None
-    current_level_index = taxonomic_levels.index(taxonomic_level)
+    current_level_index = 0  # Start from the species level
+    
+    # Determine the initial taxonomic level based on the user's decision
+    if taxonomic_decision == "No, I will let the model decide":
+        current_level_index = 0  # Start at species level if letting the model decide
+    else:
+        current_level_index = taxonomic_levels.index(taxonomic_level)  # Use specified level
 
     # Loop through taxonomic levels to check confidence
     while current_level_index < len(taxonomic_levels):