Update app.py

app.py

@@ -1,26 +1,58 @@
 import streamlit as st
 from transformers import pipeline
 from PIL import Image
+from streamlit_extras.add_vertical_space import add_vertical_space
 
 # Load a flower classification model from Hugging Face or your custom model
 flower_pipeline = pipeline(task="image-classification", model="microsoft/resnet-50")
 
-st.title("Flower Identifier 🌸")
+# Page Layout
+st.set_page_config(page_title="Flower Identifier 🌸", layout="wide", page_icon="🌼")
 
-# Upload the image
-file_name = st.file_uploader("Upload a flower image")
+# Page Header
+st.markdown(
+    """
+    <div style="text-align: center; padding: 10px;">
+        <h1 style="color: #2D6A4F; font-size: 50px;">Flower Identifier 🌸</h1>
+        <p style="color: #40916C; font-size: 20px;">Snap it, upload it, and identify the bloom!</p>
+    </div>
+    """, 
+    unsafe_allow_html=True
+)
+
+# File Upload Section
+file_name = st.file_uploader("Upload a flower image 📸", type=["jpg", "jpeg", "png"])
+
+add_vertical_space(1)
 
 if file_name is not None:
-    col1, col2 = st.columns(2)
+    col1, col2 = st.columns([1, 2])
 
-    # Display the uploaded image
+    # Display the uploaded image with shadow effect
     image = Image.open(file_name)
-    col1.image(image, use_container_width=True)
-    
-    # Make predictions
+    col1.image(
+        image, 
+        use_container_width=True, 
+        caption="Uploaded Image", 
+        output_format="auto"
+    )
+
+    # Display predictions
     predictions = flower_pipeline(image)
 
-    # Display probabilities
-    col2.header("Predictions 🌼")
+    col2.markdown("### 🌺 Predictions & Confidence Levels")
+    
     for p in predictions:
-        col2.subheader(f"{p['label']}: {round(p['score'] * 100, 1)}%")
+        col2.write(f"**{p['label']}**")
+        col2.progress(p["score"])  # Progress bar for probabilities
+
+# Footer
+st.markdown(
+    """
+    <hr style="border-top: 3px solid #40916C;">
+    <div style="text-align: center;">
+        <p style="color: #1B4332;">Powered by Hugging Face Transformers 🌿</p>
+    </div>
+    """, 
+    unsafe_allow_html=True
+)