Update app.py

app.py

@@ -48,7 +48,7 @@ def load_and_preprocess_image(image, target_size=(224, 224)):
     return img_array
 
 # Function to make predictions
-def make_prediction(image, taxonomic_level):
+def make_prediction(image, taxonomic_resolution_choice, taxonomic_level):
     # Preprocess the image
     img_array = load_and_preprocess_image(image)
 
@@ -62,7 +62,7 @@ def make_prediction(image, taxonomic_level):
     current_taxonomic_level_index = taxonomic_levels.index(taxonomic_level)
 
     # If the user chose to let the model decide, check the confidence levels
-    if taxonomic_level == "No, I will let the model decide":
+    if taxonomic_resolution_choice == "No, I will let the model decide":
         aggregated_predictions = prediction
         while current_taxonomic_level_index < len(taxonomic_levels):
             # Aggregate predictions based on the current taxonomic level