Update app2.py

app2.py

@@ -28,22 +28,27 @@ def batch_predict(data):
     if prediction.data_validation(data):
         st.success(f'Data Validation Successfull :thumbsup:')
         preds = prediction.batch_predict(data)
+        print(type(preds))
         return preds.to_csv(index=False).encode('utf-8')
     else:
         st.error(f'Data Validation Failed :thumbsdown:')
 
 st.title('Toxic Comment Classifier')
-menu = ["Single Value Prediciton","Batch Prediction"]
-choice = st.sidebar.radio("Menu",menu)
 
-if choice=="Single Value Prediciton":
+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"):
         comment = st.text_area(label="Enter your comment")
         button = st.form_submit_button(label="Submit")
         if button:
             single_predict(comment)
-else:
+#else:
+with tab2:
     st.subheader("Batch Prediction")
     csv_file = st.file_uploader("Upload File",type=['csv','parquet'])