Update app.py

app.py

@@ -51,6 +51,10 @@ def load_and_preprocess_image(image, target_size=(224, 224)):
 def make_prediction(image, taxonomic_level):
     # 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
+    
     # Make a prediction
     prediction = model.predict(img_array)