peter szemraj commited on
Commit
48b9215
1 Parent(s): b97cc67

remove brackets

Browse files
Files changed (1) hide show
  1. utils.py +8 -8
utils.py CHANGED
@@ -12,19 +12,19 @@ def postprocess(text: str):
12
  """
13
 
14
  replacements = {
15
- "ENA": "<COMPANY>",
16
- "Enron": "<COMPANY>",
17
- "Enron Corporation": "<COMPANY>",
18
- "Sony Pictures Entertainment": "<COMPANY>",
19
- "Columbia Pictures": "<COMPANY>",
20
- "Sony": "<COMPANY>",
21
- "Columbia": "<COMPANY>",
22
  "Hillary": "Jane",
23
  "Clinton": "Smith",
24
  "Amy": "Jane",
25
  "Sara": "Jane",
26
  "Harambe": "Jane",
27
- "Pascal": "<PERSON>",
28
  }
29
 
30
  # replace common values, also check lowercase
 
12
  """
13
 
14
  replacements = {
15
+ "ENA": "COMPANY",
16
+ "Enron": "COMPANY",
17
+ "Enron Corporation": "COMPANY",
18
+ "Sony Pictures Entertainment": "COMPANY",
19
+ "Columbia Pictures": "COMPANY",
20
+ "Sony": "COMPANY",
21
+ "Columbia": "COMPANY",
22
  "Hillary": "Jane",
23
  "Clinton": "Smith",
24
  "Amy": "Jane",
25
  "Sara": "Jane",
26
  "Harambe": "Jane",
27
+ "Pascal": "PERSON",
28
  }
29
 
30
  # replace common values, also check lowercase