Update app.py

app.py

@@ -39,13 +39,15 @@ results_data = {
 }
 
 # Title of the research
-st.title('Research Results')
+st.title('Variational Information bottleneck pruning for Object detection')
 
 # Create two columns with specified widths
 col1, col2 = st.columns([6, 4])
 # Right section: Filters and Hyperparameters
 with col2:
     model = st.selectbox('Select model:', models)
+   
+    st.markdown('<div class="fixed-col">', unsafe_allow_html=True)
     if model in ['SSD300', 'SSD512']:
         pruning = st.selectbox('Select pruning method:', pruning_methods)
         hyperparameter_data = hyperparameters[model][pruning]
@@ -53,10 +55,13 @@ with col2:
         dataset = st.selectbox('Select dataset:', datasets)
         hyperparameter_data = hyperparameters[model][dataset]
 
-    # Display hyperparameters table
+    st.markdown('<div class="scroller">', unsafe_allow_html=True)
     st.subheader('Hyperparameters')
+    st.markdown('<br>', unsafe_allow_html=True)  # Add space between filter and hyperparameters
     df_hyperparams = pd.DataFrame(hyperparameter_data, columns=['Model', 'kl factor', 'Pruning Epochs', 'Finetuning Epochs'])
     st.markdown(df_hyperparams.style.hide(axis="index").to_html(), unsafe_allow_html=True)
+    st.markdown('</div>', unsafe_allow_html=True)
+    st.markdown('</div>', unsafe_allow_html=True)
 
 # Left section: Results and Evolution Graphs
 with col1: