Update app.py

app.py

@@ -31,6 +31,19 @@ def main():
     st.title("Mosquito Species Classification")
     st.write("Upload a mosquito image or select an example image to classify its species.")
 
+    # File uploader for image input
+    uploaded_file = st.file_uploader("Upload your own image...", type=["jpg", "jpeg", "png"])
+
+    if uploaded_file is not None:
+        # Load the uploaded image for display
+        img = image.load_img(uploaded_file, target_size=(224, 224))
+        st.image(img, caption='Uploaded Image', width=224)
+        
+        # Classify the uploaded image
+        result, probabilities = classify_image(img)
+        st.write(f'Predicted mosquito species: **{result}**')
+        st.write(f'Prediction probabilities: {probabilities}')
+
     # Example images
     example_images = {
         "Aedes Aegypti": "Aedes_aegypti_1_0_832.jpg",
@@ -44,26 +57,13 @@ def main():
     if selected_example:
         img_path = example_images[selected_example]
         img = image.load_img(img_path, target_size=(224, 224))
-        st.image(img, width=224)
+        st.image(img, caption=f'Selected Example Image: {selected_example}', width=224)
         
         # Classify the example image
         result, probabilities = classify_image(img)
         st.write(f'Predicted mosquito species: **{result}**')
         st.write(f'Prediction probabilities: {probabilities}')
 
-    # File uploader for image input
-    uploaded_file = st.file_uploader("Or upload your own image...", type=["jpg", "jpeg", "png"])
-
-    if uploaded_file is not None:
-        # Load the image for display
-        img = image.load_img(uploaded_file, target_size=(224, 224))
-        st.image(img, caption='Uploaded Image', width=224)
-        
-        # Classify the image
-        result, probabilities = classify_image(img)
-        st.write(f'Predicted mosquito species: **{result}**')
-        st.write(f'Prediction probabilities: {probabilities}')
-
 # Run the app
 if __name__ == "__main__":
     main()