Spaces:
Runtime error
Runtime error
Commit
·
ea8d9fe
1
Parent(s):
417d05e
update
Browse files
bias_detector/bias_detector.py
CHANGED
|
@@ -42,8 +42,7 @@ class Detector:
|
|
| 42 |
f"Invalid model_type for {classifier}. Expected one of: {list(self.classifier_model_mapping[classifier].keys())}")
|
| 43 |
|
| 44 |
self.model_path = self.classifier_model_mapping[classifier][model_type]
|
| 45 |
-
|
| 46 |
-
API_token = "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"
|
| 47 |
# Using pipeline for inference
|
| 48 |
if classifier == 'Token':
|
| 49 |
task_type = "ner"
|
|
|
|
| 42 |
f"Invalid model_type for {classifier}. Expected one of: {list(self.classifier_model_mapping[classifier].keys())}")
|
| 43 |
|
| 44 |
self.model_path = self.classifier_model_mapping[classifier][model_type]
|
| 45 |
+
API_token = os.getenv("BIAS_DETECTOR_API_KEY")
|
|
|
|
| 46 |
# Using pipeline for inference
|
| 47 |
if classifier == 'Token':
|
| 48 |
task_type = "ner"
|