Update app.py

app.py

@@ -45,6 +45,11 @@ def fetch_images_from_url(url):
     img_urls = [img['src'] for img in img_tags if 'src' in img.attrs]
     return img_urls
 
+def render_intent_bars(labels, percentages):
+    for label, percentage in zip(labels, percentages):
+        bar = "#" * int(percentage // 5)  # Adjust the scale as needed
+        st.write(f"{label} | {bar} {percentage:.2f} %")
+
 def main():
     st.set_page_config(layout="wide")
     st.title("Shopping Intent Classification - SEO by DEJAN")
@@ -81,16 +86,13 @@ def main():
                     output_data_shopping_intent = run_inference(interpreter, input_data)
 
                     shopping_intent_percentages = (output_data_shopping_intent.flatten() * 100).tolist()
-                    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})")
-                        for category, percentage in zip(categories, shopping_intent_percentages):
-                            st.write(f"{category}: {percentage:.2f} %")
-                            st.progress(int(percentage))
+                        render_intent_bars(categories, shopping_intent_percentages)
                     st.write("---")
                 except Exception as e:
                     st.write(f"Could not process image {img_url}: {e}")