wu981526092 commited on
Commit
ea8d9fe
1 Parent(s): 417d05e
Files changed (1) hide show
  1. bias_detector/bias_detector.py +1 -2
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
- #API_token = os.getenv("BIAS_DETECTOR_API_KEY")
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"