Update app2.py

app2.py

@@ -16,14 +16,11 @@ prediction = model_obj()
 
 def single_predict(text):
     preds = prediction.single_predict(text)
-    col1, col2 = st.columns([1,3])
-    with col1:
-        if preds < 0.5:
-            st.success(f'Non Toxic Comment!!! :thumbsup:')
-        else:
-            st.error(f'Toxic Comment!!! :thumbsdown:')
-    with col2:
-        prediction.plot(preds)
+    if preds < 0.5:
+        st.success(f'Non Toxic Comment!!! :thumbsup:')
+    else:
+        st.error(f'Toxic Comment!!! :thumbsdown:')
+    prediction.plot(preds)
 
 def batch_predict(data):
     preds = prediction.batch_predict(data)
@@ -32,10 +29,7 @@ def batch_predict(data):
 st.title('Toxic Comment Classifier')
 st.write("This application will help to classify any comment or text in any language into 'TOXIC' or 'NON-TOXIC'")
 tab1, tab2 = st.tabs(["Single Value Prediciton","Batch Prediction"])
-#menu = ["Single Value Prediciton","Batch Prediction"]
-#choice = st.sidebar.radio("Menu",menu)
 
-#if choice=="Single Value Prediciton":
 with tab1:
     st.subheader("Prediction")
     with st.form("comment_form", clear_on_submit=True):
@@ -43,7 +37,7 @@ with tab1:
         button = st.form_submit_button(label="Submit")
         if button:
             single_predict(comment)
-#else:
+
 with tab2:
     st.subheader("Batch Prediction")
     csv_file = st.file_uploader("Upload File",type=['csv'])