Update stoneapp.py

stoneapp.py

@@ -157,19 +157,19 @@ def main():
                     st.warning("No object detected with a confidence of 0.75 or higher.")
 
     with tab2:
-        st.title('Image Classification App')
-
-        uploaded_file_1 = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"], key="file_uploader_2")
+        st.header('Image Classification')
+        uploaded_file_2 = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"], key="file_uploader_2")
         
-        if uploaded_file_1 is not None:
-            image = convert_png_to_jpg(Image.open(uploaded_file_1))
-            img, text = st.columns(2, vertical_alignment='top')
-            with img:
-                st.image(image, caption='Uploaded Image')
-            with st.spinner('Processing...'):
-                cropped_image_pil = Image.fromarray(image)
-                classification_result = classify_image(cropped_image_pil)
-                with text:
+        if uploaded_file_2 is not None:
+            image = convert_png_to_jpg(Image.open(uploaded_file_2))
+            col1, col2 = st.columns(2)
+            
+            with col1:
+                st.image(image, caption='Uploaded Image', use_column_width=True)
+            
+            with col2:
+                with st.spinner('Classifying...'):
+                    classification_result = classify_image(image)
                     if classification_result == 'pass':
                         st.success(f"Classification: {classification_result.upper()}")
                     else: