Update scoring/specificity.py
Browse files- scoring/specificity.py +3 -0
scoring/specificity.py
CHANGED
|
@@ -24,6 +24,9 @@ MODEL_NAME = os.getenv("MODEL_NAME")
|
|
| 24 |
LABEL_0 = os.getenv("LABEL_0")
|
| 25 |
LABEL_1 = os.getenv("LABEL_1")
|
| 26 |
|
|
|
|
|
|
|
|
|
|
| 27 |
# Loading the model and tokenizer
|
| 28 |
tokenizer = None
|
| 29 |
model = None
|
|
|
|
| 24 |
LABEL_0 = os.getenv("LABEL_0")
|
| 25 |
LABEL_1 = os.getenv("LABEL_1")
|
| 26 |
|
| 27 |
+
if not MODEL_NAME:
|
| 28 |
+
raise ValueError("Environment variable MODEL_NAME is not set.")
|
| 29 |
+
|
| 30 |
# Loading the model and tokenizer
|
| 31 |
tokenizer = None
|
| 32 |
model = None
|