Update app.py

app.py

@@ -51,6 +51,7 @@ variant = st.selectbox(
     index=None,
 )
 st.write("You selected model: ", variant)
+binary = st.toggle("Activate binary classification")
 st.divider()
 
 if variant == "ResNet50v2-Basemodel":
@@ -66,14 +67,10 @@ crop = st.toggle("random crop")
 compress = st.toggle("jpeg compression")
 st.divider()
 
-st.subheader("Classification")
-binary = st.toggle("Activate binary classification")
-st.divider()
-
 file_name = st.file_uploader("Choose an image...")
 
 st.button("re-run classification", type="primary")
-st.write("")
+
 
 if file_name is not None:
     col1, col2 = st.columns(2)