UI update

Hid the image as soon as it is uploaded

app.py

@@ -62,7 +62,7 @@ def main():
     if uploaded_file is not None:
         # Display uploaded image
         image = Image.open(io.BytesIO(uploaded_file.read()))
-        st.image(image, caption="Uploaded Image", use_column_width=True)
+        # st.image(image, caption="Uploaded Image", use_column_width=True)
         
         # Load model
         model = load_model()
@@ -76,7 +76,7 @@ def main():
         )
         if predict_button:
             # Display processing message with spinner
-            with st.spinner("Please wait...Processing the image and predicting..."):
+            with st.spinner(" Please wait... Processing the image and predicting..."):
                 # Preprocess image
                 processed_image = preprocess_image(image)
                 
@@ -94,7 +94,7 @@ def main():
                     st.image(explanation_image, caption="Explanation Image", use_column_width=True)
                 
                 # Display prediction
-                st.subheader("Prediction:")
+                st.subheader("Prediction")
                 st.write(f"Predicted Class: {predicted_class}")
                 st.write(f"Confidence Score: {confidence_score}%")