Update src/streamlit_app.py

src/streamlit_app.py

@@ -333,8 +333,8 @@ def create_performance_chart(df):
     return fig
 
 # Enhanced model details display function
-def display_model_categories(models_dict: Dict[str, List[Dict]], ram_gb: int):
-    """Display models organized by category with download sizes"""
+def display_model_categories(models_dict: Dict[str, List[Dict]], ram_gb: int, show_quantization=True):
+    """Display models with quantization options"""
     if not models_dict:
         return
     
@@ -343,14 +343,25 @@ def display_model_categories(models_dict: Dict[str, List[Dict]], ram_gb: int):
     for category, model_list in models_dict.items():
         if model_list:
             with st.expander(f"📂 {category.replace('_', ' ').title()} Models"):
-                for model in model_list[:8]:  # Limit to top 8 per category
-                    col1, col2, col3 = st.columns([3, 1, 2])
-                    with col1:
-                        st.markdown(f"**{model['name']}**")
-                    with col2:
-                        st.markdown(f"{model['size']}")
-                    with col3:
-                        st.markdown(f"*{model['description']}*")
+                for model in model_list[:6]:  # Reduced to show quantization options
+                    st.markdown(f"**{model['name']}**")
+                    st.markdown(f"*{model['description']}*")
+                    
+                    if show_quantization:
+                        # Create quantization size table
+                        quant_cols = st.columns(4)
+                        for i, (quant_type, quant_info) in enumerate(QUANTIZATION_FORMATS.items()):
+                            with quant_cols[i]:
+                                quant_size = calculate_quantized_size(model['size'], quant_type)
+                                st.metric(
+                                    label=quant_type,
+                                    value=quant_size,
+                                    help=quant_info['description']
+                                )
+                    else:
+                        st.markdown(f"**Original Size:** {model['size']}")
+                    
+                    st.markdown("---")
 
 # Demo data generator for when Excel files are not available
 def generate_demo_data():