Update app.py

app.py

@@ -199,21 +199,21 @@ def main():
         st.image(image, width=150) #,use_column_width=True)
         page = option_menu(menu_title='Menu',
                         menu_icon="robot",
-                        options=["Car Parts Detection",
+                        options=["Damage Detection",
                                     "Under Construction"],
-                        icons=["chat-dts",
+                        icons=["camera",
                                 "key"],
                         default_index=0
                         )
 
         # Additional section below the option menu
         # st.markdown("---")  # Add a separator line
-        st.header("Settings")
+        # st.header("Settings")
         
     st.title('ITACA Insurance Core AI Module')
 
-    if page == "Car Parts Detection":
-        st.header('Car Parts Detection')
+    if page == "Damage Detection":
+        st.header('Car Parts Damage Detection')
 
         st.write(
             """
@@ -232,14 +232,14 @@ def main():
             st.write("Please upload an image.")
 
         if st.button("Prediction"):
-            # Call the inference function with the uploaded image
-            imagen1, imagen2, imagen3, partes = inference(image)
-    
-            st.image(imagen1, caption="crash image1")
-            st.image(imagen2, caption="crash image2")
-            st.image(imagen3, caption="crash image3")
-            st.write(partes)
-            print(partes)
+            with st.spinner("Loading..."):
+                # Call the inference function with the uploaded image
+                imagen1, imagen2, imagen3, partes = inference(image)
+        
+                st.image(imagen1, caption="crash image1")
+                st.image(imagen2, caption="crash image2")
+                st.image(imagen3, caption="crash image3")
+                st.table(partes)
 
     elif page == "Under Construction":
         st.header('Under Construction')