Update predict.py
Browse files- scripts/predict.py +2 -2
scripts/predict.py
CHANGED
@@ -11,8 +11,8 @@ from transformers import (
|
|
11 |
AutoModelForSequenceClassification
|
12 |
)
|
13 |
|
14 |
-
BIN_REPO = 'dlsmallw/Binary-Classification
|
15 |
-
ML_REPO = 'dlsmallw/Multilabel-Regression
|
16 |
|
17 |
## Class used to encapsulate and handle the logic for inference
|
18 |
class InferenceHandler:
|
|
|
11 |
AutoModelForSequenceClassification
|
12 |
)
|
13 |
|
14 |
+
BIN_REPO = 'dlsmallw/NLPinitiative-Binary-Classification'
|
15 |
+
ML_REPO = 'dlsmallw/NLPinitiative-Multilabel-Regression'
|
16 |
|
17 |
## Class used to encapsulate and handle the logic for inference
|
18 |
class InferenceHandler:
|