Update src/app.py

Fix columns, hide graph by default

src/app.py

@@ -186,7 +186,8 @@ col1, col2 = st.columns([1,1.3])
 if gpu_vendor == "Apple":
     col.warning("""For M1/M2 Apple chips, PyTorch uses [Metal Performance Shaders (MPS)](https://huggingface.co/docs/accelerate/usage_guides/mps) as backend.\\
 Remember that Apple M1/M2 chips share memory between CPU and GPU.""", icon="⚠️")
-with col1:
+_, col, _ = st.columns([1,3,1])
+with col:
     st.write(f"####  [{model_name}](https://huggingface.co/{model_name}) ({custom_ceil(memory_table.iloc[3,0],1):.1f}B)")
 
     dtypes = memory_table.columns.tolist()[::-1]
@@ -199,7 +200,8 @@ with col1:
             info = _memory_table[_memory_table['dtype'] == dtype].set_index('Variable')
             show_gpu_info(info, lora_pct, gpu_vendor)
     st.write(memory_table.iloc[[0, 1, 2, 4]])
-with col1:
+_, col, _ = st.columns([1,3,1])
+with col.expander("Graph", expanded=False):
     extra = ""
     if gpu_vendor == "Apple":
         st.warning("This graph is irrelevant for M1/M2 chips as they can't run in parallel.", icon="⚠️")