Sephfox commited on
Commit
d07dc04
·
verified ·
1 Parent(s): 9473e59

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ from sklearn.model_selection import train_test_split
14
  from sklearn.preprocessing import OneHotEncoder
15
  from sklearn.neural_network import MLPClassifier
16
  from deap import base, creator, tools, algorithms
17
- from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
18
  import gc
19
  import multiprocessing as mp
20
  from joblib import Parallel, delayed
 
14
  from sklearn.preprocessing import OneHotEncoder
15
  from sklearn.neural_network import MLPClassifier
16
  from deap import base, creator, tools, algorithms
17
+ from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline, AutoModelForSequenceClassification
18
  import gc
19
  import multiprocessing as mp
20
  from joblib import Parallel, delayed