Spaces:
Running
Running
Update app.py
Browse files
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
|