shivansh-ka commited on
Commit
21fccbb
·
1 Parent(s): 34c68b5

Update app2.py

Browse files
Files changed (1) hide show
  1. app2.py +6 -9
app2.py CHANGED
@@ -26,15 +26,12 @@ def single_predict(text):
26
  prediction.plot(preds)
27
 
28
  def batch_predict(data):
29
- if prediction.data_validation(data):
30
- print("batch in")
31
- st.success(f'Data Validation Successfull :thumbsup:')
32
- preds = prediction.batch_predict(data)
33
- print(type(preds))
34
- return preds.to_csv(index=False).encode('utf-8')
35
- else:
36
- st.error(f'Data Validation Failed :thumbsdown:')
37
-
38
  st.title('Toxic Comment Classifier')
39
 
40
  tab1, tab2 = st.tabs(["Single Value Prediciton","Batch Prediction"])
 
26
  prediction.plot(preds)
27
 
28
  def batch_predict(data):
29
+ print("batch in")
30
+ st.success(f'Data Validation Successfull :thumbsup:')
31
+ preds = prediction.batch_predict(data)
32
+ print(type(preds))
33
+ return preds.to_csv(index=False).encode('utf-8')
34
+
 
 
 
35
  st.title('Toxic Comment Classifier')
36
 
37
  tab1, tab2 = st.tabs(["Single Value Prediciton","Batch Prediction"])