Update app.py

app.py

@@ -48,12 +48,18 @@ 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_decision, taxonomic_level):
+    # Determine the taxonomic level based on the user's decision
+    if taxonomic_decision == "Yes, I want to specify the taxonomic level":
+        taxonomic_level = taxonomic_level  # Use the level specified in the dropdown
+    else:
+        taxonomic_level = "species"  # Default to species if not specified
+    
     # Preprocess the image
     img_array = load_and_preprocess_image(image)
     
     # Get the class names from the 'species' column
-    class_names = sorted(taxo_df['species'].unique())  # Add this line to define class_names
+    class_names = sorted(taxo_df['species'].unique())
     
     # Make a prediction
     prediction = model.predict(img_array)