Update app.py

app.py

@@ -6,7 +6,6 @@ import requests
 from io import BytesIO
 from bs4 import BeautifulSoup
 import pandas as pd
-import base64
 import os
 
 def download_model(model_url, model_path):
@@ -66,9 +65,6 @@ def main():
             input_shape = input_details[0]['shape']
             input_size = input_shape[1]  # assuming square input
 
-            data = []
-            errors = []
-
             for img_url in img_urls:
                 try:
                     response = requests.get(img_url)
@@ -78,31 +74,20 @@ def main():
                     output_data_shopping_intent = run_inference(interpreter, input_data)
 
                     shopping_intent_percentages = (output_data_shopping_intent.flatten() * 100).tolist()
-
-                    data.append({
-                        'URL': img_url,
-                        'Shopping Intent 1 (%)': shopping_intent_percentages[0],
-                        'Shopping Intent 2 (%)': shopping_intent_percentages[1],
-                        'Shopping Intent 3 (%)': shopping_intent_percentages[2],
-                        'Shopping Intent 4 (%)': shopping_intent_percentages[3]
-                    })
-
-                    # Display each image and its classification
-                    st.image(image, caption=f"URL: {img_url}", width=100)
-                    st.write(f"Shopping Intent 1 (%): {shopping_intent_percentages[0]}")
-                    st.write(f"Shopping Intent 2 (%): {shopping_intent_percentages[1]}")
-                    st.write(f"Shopping Intent 3 (%): {shopping_intent_percentages[2]}")
-                    st.write(f"Shopping Intent 4 (%): {shopping_intent_percentages[3]}")
+                    formatted_percentages = [f"{intent:.2f} %" for intent in shopping_intent_percentages]
+
+                    col1, col2 = st.columns([1, 3])
+                    with col1:
+                        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]}")
                     st.write("---")
-
                 except Exception as e:
-                    errors.append(f"Could not process image {img_url}: {e}")
-
-            # Display errors in an expandable section
-            if errors:
-                with st.expander(f"Could not process {len(errors)} images"):
-                    for error in errors:
-                        st.write(error)
+                    st.write(f"Could not process image {img_url}: {e}")
 
 if __name__ == "__main__":
     main()