improve the way results are displayed

src/app.py

@@ -22,6 +22,24 @@ def get_gpu_specs():
     return pd.read_csv("data/gpu_specs.csv")
 
 
+def show_gpu_info(info, trainable_params=0):
+    for var in ['Inference', 'Full Training Adam', 'LoRa Fine-tuning']:
+        _info = info.loc[var]
+        if _info['Number of GPUs'] >= 3:
+            func = st.error
+            icon = "⛔"
+        elif _info['Number of GPUs'] == 2:
+            func = st.warning
+            icon = "⚠️"
+        else:
+            func = st.success
+            icon = "✅"
+        
+        msg = f"You require **{_info['Number of GPUs']}** GPUs for **{var}**"
+        if var == 'LoRa Fine-tuning':
+            msg += f" ({trainable_params}%)"
+        func(msg, icon=icon)
+
 
 def get_name(index):
     row = gpu_specs.iloc[index]
@@ -93,10 +111,17 @@ _memory_table = _memory_table.apply(np.ceil).astype(int).drop(columns=['Paramete
 _memory_table.columns = ['Inference', 'Full Training Adam', 'LoRa Fine-tuning']
 _memory_table = _memory_table.stack().reset_index()
 _memory_table.columns = ['dtype', 'Variable', 'Number of GPUs']
-
+st.write()
 col1, col2 = st.columns([1,1.3])
 with col1:
     st.write(f"####  [{model_name}](https://huggingface.co/{model_name}) ({memory_table.iloc[3,0]:.1f}B)")
+
+    dtypes = memory_table.columns.tolist()[::-1]
+    tabs = st.tabs(dtypes)
+    for dtype, tab in zip(dtypes, tabs):
+        with tab:
+            info = _memory_table[_memory_table['dtype'] == dtype].set_index('Variable')
+            show_gpu_info(info, lora_pct)
     st.write(memory_table.iloc[[0, 1, 2, 4]])
 with col2:
     num_colors= 4
@@ -106,3 +131,4 @@ with col2:
                     , xaxis_tickfont_size=14, yaxis_tickfont_size=16, yaxis_dtick='1')
     st.plotly_chart(fig, use_container_width=True)
 
+