Update app.py

app.py

@@ -21,27 +21,22 @@ def get_result(img_path):
     prediction = model.predict(img_input)
 
     if prediction[0][0] < 0.5:
-        return "He is a Men."
+        return "He is a Men 🚹"
     else:
-        return "She is a Women."
+        return "She is a Women 🚺"
     
            
-# Set the title of the app
-st.title('Image Input and Display')
+st.title('Let's detect the gender 🚹🚺')
 
-# Upload image
 uploaded_image = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
 
-# If an image is uploaded, display it along with text
 if uploaded_image is not None:
-    # Open the image using PIL
-    
-    # output = get_result(uploaded_image)
-    
+
     image = Image.open(uploaded_image)
     
     image_path = os.path.join(upload_folder, uploaded_image.name)
     image.save(image_path)
     output = get_result(image_path)
-    # Display the image
-    st.image(image, caption= output, use_container_width=True)
+    
+    st.write(output)
+    st.image(image, use_container_width=True)