Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -13,7 +13,7 @@ from huggingface_hub import hf_hub_download, login
|
|
13 |
import model_archs
|
14 |
from model_archs import BertClassifier, LogisticRegressionTorch, SimpleCNN, MLP, Pool2BN
|
15 |
import tangermeme
|
16 |
-
from tangermeme import one_hot_encode
|
17 |
|
18 |
# Load label mapping
|
19 |
label_to_int = pd.read_pickle('label_to_int.pkl')
|
|
|
13 |
import model_archs
|
14 |
from model_archs import BertClassifier, LogisticRegressionTorch, SimpleCNN, MLP, Pool2BN
|
15 |
import tangermeme
|
16 |
+
from tangermeme.utils import one_hot_encode
|
17 |
|
18 |
# Load label mapping
|
19 |
label_to_int = pd.read_pickle('label_to_int.pkl')
|