Update app.py

app.py

@@ -1,25 +1,22 @@
 import streamlit as st
 from fastai.vision.all import *
 from PIL import Image
-
-# Load the trained model
 learn = load_learner('export.pkl')
 
-# Define a function to make predictions on an image
 def predict_image(image):
     img = PILImage.create(image)
     pred, pred_idx, probs = learn.predict(img)
     return pred, probs[pred_idx].item()
 
-# Main function for Streamlit app
+
 def main():
-    st.title("Image Classifier")
+    st.title(" Gender Image Classifier")
     st.write("Upload an image to classify")
 
-    # Upload image file
+   
     uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "png", "jpeg"])
 
-    # Display prediction when an image is uploaded
+   
     if uploaded_file is not None:
         image = Image.open(uploaded_file)
         st.image(image, caption='Uploaded Image.', use_column_width=True)
@@ -27,6 +24,6 @@ def main():
         prediction, probability = predict_image(uploaded_file)
         st.write(f"Prediction: {prediction}; Probability: {probability:.4f}")
 
-# Run the Streamlit app
+
 if __name__ == '__main__':
     main()