Update src/streamlit_app.py

src/streamlit_app.py

@@ -161,7 +161,7 @@ with col_right:
                 with tab1:
                     with st.container(height=300):
                         fw = fusion_weights[0].cpu().numpy()
-                        fig, ax = plt.subplots(figsize=(9.5, 2.4))
+                        fig, ax = plt.subplots(figsize=(13.5, 2.4))
                         labels = ['Tabular', 'Image']
                         values = [fw[1], fw[0]]
                         ax.barh(labels, values)
@@ -182,7 +182,7 @@ with col_right:
                             inv_map = {v: k for k, v in class_map.items()}
                             p_labels = [inv_map[i] for i in range(len(p_vals))]
                 
-                        fig2, ax2 = plt.subplots(figsize=(8.5, 1.4))
+                        fig2, ax2 = plt.subplots(figsize=(11.5, 1.4))
                         ax2.bar(p_labels, p_vals)
                         ax2.set_ylim(0, 1)
                         ax2.set_ylabel("Probability")