made table sizing a little smarter

app.py

@@ -8,6 +8,8 @@ from st_aggrid import AgGrid, GridOptionsBuilder, JsCode
 from os.path import exists
 import threading
 
+st.set_page_config(layout="wide")
+
 
 def get_model_infos():
     api = HfApi()
@@ -206,7 +208,8 @@ if len(dataset_df) > 0:
     )
 
     go = gb.build()
-    AgGrid(dataset_df, gridOptions=go, allow_unsafe_jscode=True)
+    fit_columns = len(dataset_df.columns) < 10
+    AgGrid(dataset_df, gridOptions=go, height=28*len(dataset_df) + (35 if fit_columns else 41), allow_unsafe_jscode=True, fit_columns_on_grid_load=fit_columns)
 
 else:
     st.markdown(