Spaces:
Running
Running
Update predictor.py
Browse files- predictor.py +1 -1
predictor.py
CHANGED
@@ -4,7 +4,7 @@ import warnings
|
|
4 |
warnings.filterwarnings('ignore')
|
5 |
import logging
|
6 |
logging.disable(logging.WARNING)
|
7 |
-
from
|
8 |
import torch
|
9 |
import numpy as np
|
10 |
import random
|
|
|
4 |
warnings.filterwarnings('ignore')
|
5 |
import logging
|
6 |
logging.disable(logging.WARNING)
|
7 |
+
from Mbert import SentencePairDataset, MBERTClassifier
|
8 |
import torch
|
9 |
import numpy as np
|
10 |
import random
|