thotran commited on
Commit
6423daf
1 Parent(s): 9993f32

rm error-badline

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -25,10 +25,10 @@ BATCH_SIZE = 32
25
  PRE_TRAINED_MODEL_NAME = "distilbert-base-uncased"
26
 
27
  #import all data
28
- data=pd.read_csv('./data/train.csv',engine='python',encoding='utf-8', error_bad_lines=False)
29
- test=pd.read_csv('./data/test.csv',engine='python',encoding='utf-8', error_bad_lines=False)
30
- test_labels=pd.read_csv('./data/test_labels.csv',engine='python',encoding='utf-8', error_bad_lines=False)
31
- sub=pd.read_csv('./data/sample_submission.csv',engine='python',encoding='utf-8', error_bad_lines=False)
32
 
33
  #setup data
34
  data.drop(columns='id',inplace=True)
@@ -217,6 +217,6 @@ model = model.to(device)
217
  #sub.insert(1,"tweet",data.comment_text,True)
218
  #sub.to_csv("sub.csv", encoding='utf-8', index=False)
219
  #^commented above code, saved to csv to reduce wait/comput time
220
- sub=pd.read_csv('./data/sub.csv',engine='python',encoding='utf-8', error_bad_lines=False)
221
  sub.drop(index="id")
222
  st.dataframe(sub)
 
25
  PRE_TRAINED_MODEL_NAME = "distilbert-base-uncased"
26
 
27
  #import all data
28
+ data=pd.read_csv('./data/train.csv',engine='python',encoding='utf-8')
29
+ test=pd.read_csv('./data/test.csv',engine='python',encoding='utf-8')
30
+ test_labels=pd.read_csv('./data/test_labels.csv',engine='python',encoding='utf-8')
31
+ sub=pd.read_csv('./data/sample_submission.csv',engine='python',encoding='utf-8')
32
 
33
  #setup data
34
  data.drop(columns='id',inplace=True)
 
217
  #sub.insert(1,"tweet",data.comment_text,True)
218
  #sub.to_csv("sub.csv", encoding='utf-8', index=False)
219
  #^commented above code, saved to csv to reduce wait/comput time
220
+ sub=pd.read_csv('./data/sub.csv',engine='python',encoding='utf-8')
221
  sub.drop(index="id")
222
  st.dataframe(sub)