try to fix overlapping cols

app.py

@@ -24,45 +24,47 @@ if method_selection == METHOD_OPTIONS["nsp"]:
 
 st.header("Configure prompts and labels")
 col1, col2 = st.columns(2)
-col1.subheader("Candidate labels")
-labels = col1.text_area(
-    label="These are the labels that the model will try to predict for the given text input. Your input labels should be comma separated and meaningful.",
-    value="spor,dünya,siyaset,ekonomi,kültür ve sanat",
-)
-col2.subheader("Prompt template")
-prompt_template = col2.text_area(
-    label="Prompt template is used to transform NLI and NSP tasks into a general-use zero-shot classifier. Models replace {} with the labels that you have given.",
-    value="Bu metin {} kategorisine aittir",
-)
 
-col1.header("Make predictions")
-col2.header("")
-col1.text_area("", value="Enter some text to classify.")
-col1.button("Predict")
+with col1:
+    st.subheader("Candidate labels")
+    labels = st.text_area(
+        label="These are the labels that the model will try to predict for the given text input. Your input labels should be comma separated and meaningful.",
+        value="spor,dünya,siyaset,ekonomi,kültür ve sanat",
+    )
+    st.header("Make predictions")
+    st.text_area("", value="Enter some text to classify.")
+    st.button("Predict")
 
-probs = [0.86, 0.10, 0.01, 0.02, 0.01]
-data = pd.DataFrame({"labels": labels.split(","), "probability": probs}).sort_values(
-    by="probability", ascending=False
-)
-chart = px.bar(
-    data,
-    x="probability",
-    y="labels",
-    color="labels",
-    orientation="h",
-    height=290,
-    width=500,
-).update_layout(
-    {
-        "xaxis": {"title": "probability", "visible": True, "showticklabels": True},
-        "yaxis": {"title": None, "visible": True, "showticklabels": True},
-        "margin": dict(
-            l=10,  # left
-            r=10,  # right
-            t=50,  # top
-            b=10,  # bottom
-        ),
-        "showlegend": False,
-    }
-)
-col2.plotly_chart(chart)
+with col2:
+    st.subheader("Prompt template")
+    prompt_template = st.text_area(
+        label="Prompt template is used to transform NLI and NSP tasks into a general-use zero-shot classifier. Models replace {} with the labels that you have given.",
+        value="Bu metin {} kategorisine aittir",
+    )
+    st.header("")
+    probs = [0.86, 0.10, 0.01, 0.02, 0.01]
+    data = pd.DataFrame(
+        {"labels": labels.split(","), "probability": probs}
+    ).sort_values(by="probability", ascending=False)
+    chart = px.bar(
+        data,
+        x="probability",
+        y="labels",
+        color="labels",
+        orientation="h",
+        height=290,
+        width=500,
+    ).update_layout(
+        {
+            "xaxis": {"title": "probability", "visible": True, "showticklabels": True},
+            "yaxis": {"title": None, "visible": True, "showticklabels": True},
+            "margin": dict(
+                l=10,  # left
+                r=10,  # right
+                t=50,  # top
+                b=10,  # bottom
+            ),
+            "showlegend": False,
+        }
+    )
+    st.plotly_chart(chart)