Update app.py

app.py

@@ -82,6 +82,10 @@ def make_prediction(image, taxonomic_decision, taxonomic_level):
         
         current_level_index += 1  # Move to the next taxonomic level
 
+    # Check if a valid prediction was made
+    if current_level_index == len(taxonomic_levels):
+        return "Unknown animal"  # No valid predictions met the confidence criteria
+    
     # Get the predicted class name for the top prediction
     predicted_class_index = np.argmax(aggregated_predictions)
     predicted_class_name = aggregated_class_labels[predicted_class_index]