Upload 3 files

app.py

@@ -0,0 +1,96 @@
+import streamlit as st
+from tensorflow.keras.models import load_model
+from PIL import Image
+import numpy as np
+import matplotlib.pyplot as plt
+import seaborn as sns
+
+# Load the trained model
+model = load_model('cnn_model.h5')
+
+# Function to process the uploaded image
+def process_image(img):
+    img = img.convert('RGB')  
+    img = img.resize((32, 32)) 
+    img = np.array(img)  
+    img = img / 255.0  
+    img = np.expand_dims(img, axis=0) 
+    return img
+
+# Frontend design
+st.set_page_config(page_title="Cassava Leaf Disease Image Classification", layout="wide")
+st.title("Cassava Leaf Disease Image Classification")
+st.markdown("""
+This web application helps in classifying cassava leaf diseases based on an uploaded image. 
+Simply upload a cassava leaf image, and the model will predict the disease and display the probabilities for each class.
+""")
+
+# Image upload
+file = st.file_uploader('Select an image (JPG, JPEG, PNG)', type=['jpg', 'jpeg', 'png'])
+
+if file is not None:
+    # Display uploaded image
+    img = Image.open(file)
+    st.image(img, caption='Uploaded Image', use_column_width=True)
+
+    # Preprocess the image
+    image = process_image(img)
+    
+    # Model prediction
+    with st.spinner('Classifying the image...'):
+        predictions = model.predict(image)
+        predicted_class = np.argmax(predictions)  
+        predicted_prob = predictions[0][predicted_class]  
+
+    # Class names for prediction
+    class_names = ["Cassava Bacterial Blight (CBB)", "Cassava Brown Streak Disease (CBSD)", 
+                   "Cassava Green Mottle (CGM)", "Cassava Mosaic Disease (CMD)", "Healthy"]
+
+    # Display the prediction result
+    st.subheader(f"**Prediction:** {class_names[predicted_class]}")
+    st.write(f"**Confidence:** {predicted_prob * 100:.2f}%")
+
+    # Display prediction probabilities
+    st.write("**Prediction Probabilities for Each Class:**")
+
+    # Prepare probabilities for visualization
+    probabilities = predictions[0]
+    prob_dict = {class_names[i]: probabilities[i] for i in range(len(class_names))}
+    
+    # Plot settings
+    sns.set(style="whitegrid", palette="muted")  # Use a muted color palette for the plot
+
+    # Create the figure for the bar chart
+    fig, ax = plt.subplots(figsize=(12, 6))  # Adjust figure size for better readability
+
+    # Plot the bar chart with a color palette
+    ax.bar(list(prob_dict.keys()), list(prob_dict.values()), color=sns.color_palette("Blues", len(class_names)), edgecolor='black')
+    ax.set_ylabel('Probability', fontsize=14, color='black')
+    ax.set_title('Prediction Probabilities for Each Class', fontsize=18, color='black')
+
+    # Rotate x-axis labels for better readability
+    plt.xticks(rotation=45, ha='right', fontsize=12)
+
+    # Annotate bars with percentage values
+    for index, value in enumerate(prob_dict.values()):
+        ax.text(index, value, f'{value * 100:.0f}%', va='bottom', ha='center', fontsize=10, color='black')
+
+    # Style improvements: Remove background grid and spines
+    ax.spines['top'].set_visible(False)
+    ax.spines['right'].set_visible(False)
+    ax.spines['left'].set_visible(False)
+    ax.spines['bottom'].set_visible(False)
+    ax.grid(False)
+
+    # Adjust layout to prevent clipping
+    fig.tight_layout()
+
+    # Display the plot in Streamlit
+    st.pyplot(fig)
+
+# Footer section
+st.markdown("""
+---  
+**Cassava Leaf Disease Image Classification** | Developed with ❤️ by Senasu Demir  
+For more information, visit [Your Website or Contact Info].
+""")

cnn_model.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4d4c3a7096e7d1ef8546a3be37880878faae5529d1d15c3b8549831b45747666
+size 96313928

requirements.txt

@@ -0,0 +1,5 @@
+streamlit
+tensorflow
+Pillow
+matplotlib
+seaborn