Update app.py

app.py

@@ -3,47 +3,6 @@ from PIL import Image
 import numpy as np
 from tensorflow.keras.models import load_model
 
-# Custom CSS for styling
-custom_css = """
-<style>
-body {
-    background-color: #f0f2f6;
-}
-.stButton>button {
-    background-color: #4CAF50;
-    color: white;
-    padding: 10px 15px;
-    text-align: center;
-    text-decoration: none;
-    display: inline-block;
-    font-size: 16px;
-    margin: 4px 2px;
-    cursor: pointer;
-    border-radius: 8px;
-    border: none;
-}
-.stButton>button:hover {
-    background-color: #45a049;
-}
-.stTextInput>div>div>input {
-    border-radius: 8px;
-}
-.stTextInput>div>div>input:focus {
-    border-color: #4CAF50;
-    box-shadow: none;
-}
-</style>
-"""
-
-st.markdown(custom_css, unsafe_allow_html=True)
-
-# Set page configuration
-st.set_page_config(
-    page_title="Face Mask Detection App",
-    page_icon=":mask:",
-    layout="wide"
-)
-
 # Define a function to make predictions on a single image
 def predict_single_image(image, model):
     try:
@@ -81,9 +40,38 @@ def main():
     st.title("Face Mask Detection App")
     st.write("Upload an image and click 'Predict' to see the prediction.")
 
-    # Header Image
-    header_image = Image.open("header_image.jpg")
-    st.image(header_image, use_column_width=True)
+    # Custom CSS for styling
+    custom_css = """
+    <style>
+    body {
+        background-color: #f0f2f6;
+    }
+    .stButton>button {
+        background-color: #4CAF50;
+        color: white;
+        padding: 10px 15px;
+        text-align: center;
+        text-decoration: none;
+        display: inline-block;
+        font-size: 16px;
+        margin: 4px 2px;
+        cursor: pointer;
+        border-radius: 8px;
+        border: none;
+    }
+    .stButton>button:hover {
+        background-color: #45a049;
+    }
+    .stTextInput>div>div>input {
+        border-radius: 8px;
+    }
+    .stTextInput>div>div>input:focus {
+        border-color: #4CAF50;
+        box-shadow: none;
+    }
+    </style>
+    """
+    st.markdown(custom_css, unsafe_allow_html=True)
 
     uploaded_file = st.file_uploader("", type=["jpg", "png", "jpeg"])