Spaces:
Runtime error
Runtime error
wu981526092
commited on
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"
|