Update app.py

app.py

@@ -44,6 +44,23 @@ def display_diff(diff):
     
     return left_html, right_html, diff_found
 
+# Apply custom CSS for wider layout
+st.markdown(
+    """
+    <style>
+    .reportview-container .main .block-container {
+        max-width: 100%;
+        padding-left: 0%;
+        padding-right: 0%;
+    }
+    .stMarkdown {
+        white-space: pre-wrap;
+    }
+    </style>
+    """,
+    unsafe_allow_html=True
+)
+
 st.title("Model Structure Comparison Tool")
 model_id1 = st.text_input("Enter the first HuggingFace Model ID")
 model_id2 = st.text_input("Enter the second HuggingFace Model ID")
@@ -68,20 +85,3 @@ if model_id1 and model_id2:
     with col2:
         st.write("### Model 2")
         st.markdown(right_html, unsafe_allow_html=True)
-
-# Apply custom CSS for wider layout
-st.markdown(
-    """
-    <style>
-    .reportview-container .main .block-container {
-        max-width: 90%;
-        padding-left: 5%;
-        padding-right: 5%;
-    }
-    .stMarkdown {
-        white-space: pre-wrap;
-    }
-    </style>
-    """,
-    unsafe_allow_html=True
-)