Update app.py

app.py

@@ -65,6 +65,13 @@ def main():
             input_shape = input_details[0]['shape']
             input_size = input_shape[1]  # assuming square input
 
+            categories = [
+                "No Shopping Intent",
+                "Fashion & Style",
+                "Home & Garden",
+                "Tools, Vehicles, Electronics & Appliances"
+            ]
+
             for img_url in img_urls:
                 try:
                     response = requests.get(img_url)
@@ -81,10 +88,9 @@ def main():
                         st.image(image.resize((224, 224)), width=224)
                     with col2:
                         st.write(f"[URL]({img_url})")
-                        st.write(f"Intent 1: {formatted_percentages[0]}")
-                        st.write(f"Intent 2: {formatted_percentages[1]}")
-                        st.write(f"Intent 3: {formatted_percentages[2]}")
-                        st.write(f"Intent 4: {formatted_percentages[3]}")
+                        for category, percentage in zip(categories, shopping_intent_percentages):
+                            st.write(f"{category}: {percentage:.2f} %")
+                            st.progress(int(percentage))
                     st.write("---")
                 except Exception as e:
                     st.write(f"Could not process image {img_url}: {e}")