Spaces:
Sleeping
Sleeping
Update processor_bert.py
Browse files- processor_bert.py +1 -1
processor_bert.py
CHANGED
|
@@ -25,7 +25,7 @@ _load_lock = threading.Lock()
|
|
| 25 |
MODEL_PATH = os.path.join(os.path.dirname(__file__), 'models', 'log_classifier.joblib')
|
| 26 |
ONNX_DIR = os.path.join(os.path.dirname(__file__), 'models', 'onnx')
|
| 27 |
CONFIDENCE_THRESHOLD = 0.30
|
| 28 |
-
DEFAULT_BATCH =
|
| 29 |
|
| 30 |
|
| 31 |
def preload_models():
|
|
|
|
| 25 |
MODEL_PATH = os.path.join(os.path.dirname(__file__), 'models', 'log_classifier.joblib')
|
| 26 |
ONNX_DIR = os.path.join(os.path.dirname(__file__), 'models', 'onnx')
|
| 27 |
CONFIDENCE_THRESHOLD = 0.30
|
| 28 |
+
DEFAULT_BATCH = 512
|
| 29 |
|
| 30 |
|
| 31 |
def preload_models():
|