Hugging Face's logo

Spaces:
AIEcosystem
/
Multilingual
Sleeping

App Files Community
AIEcosystem commited on
Commit
6303d91
·
verified ·
1 Parent(s): 57534fb

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +41 -44
src/streamlit_app.py CHANGED
@@ -108,58 +108,55 @@ if not comet_initialized:
108
  # --- Label Definitions ---
109
 
110
  labels = [
111
- "prefix", "person name",
112
- "job_title", "company_name", "job_area",
113
- "email_address",
114
- "account_number", "address", "city", "postal_code",
115
- "password", "job_type", "state",
116
- "county",
117
- "phone_number",
118
- "personal identification_number",
119
- "gender", "biological_sex",
120
- "nationality",
121
- "medical_condition", "age",
 
 
122
 
123
  ]
124
 
125
 
126
  # Create a mapping dictionary for labels to categories
127
 
128
- category_mapping = {
129
-
 
 
 
 
 
130
 
131
- "Personal Information": [
132
- "person_name",
133
- "prefix",
134
- "gender",
135
- "biological_sex",
136
- "age"
137
- ],
138
- "Contact Information": [
139
- "email_address",
140
- "address",
141
- "city",
142
- "state",
143
- "county",
144
- "postal_code",
145
- "phone_number"
146
- ],
147
- "Professional Information": [
148
- "job_title",
149
- "job_area",
150
- "job_type",
151
- "company_name"
152
- ],
153
- "Sensitive Information": [
154
- "password",
155
- "personal_identification_number",
156
-
157
- "nationality",
158
-
159
- "medical_condition",
160
- "account_number"
161
- ]
162
  }
 
 
163
 
164
 
165
 
 
108
  # --- Label Definitions ---
109
 
110
  labels = [
111
+ "PERSON",
112
+ "FIRST_NAME",
113
+
114
+ "LAST_NAME",
115
+ ,
116
+ "TITLE", "JOB_TITLE",
117
+ "AFFILIATION", "GENDER",
118
+ "AGE",
119
+ "DATE_OF_BIRTH",
120
+ "PLACE_OF_BIRTH",
121
+ "NATIONALITY", "LOCATION",
122
+ "CITY", "COUNTRY", "ROLE",
123
+ "RELATIONSHIP"
124
 
125
  ]
126
 
127
 
128
  # Create a mapping dictionary for labels to categories
129
 
130
+ category_mapping =
131
+ {
132
+ "Identity": [
133
+ "PERSON",
134
+ "FIRST_NAME",
135
+
136
+ "LAST_NAME",
137
 
138
+ "TITLE"
139
+ ],
140
+ "Professional": [
141
+ "JOB_TITLE",
142
+ "AFFILIATION"
143
+ ],
144
+ "Demographic": [
145
+ "GENDER",
146
+ "AGE",
147
+ "DATE_OF_BIRTH",
148
+ "PLACE_OF_BIRTH",
149
+ "NATIONALITY",
150
+ "LOCATION",
151
+ "CITY", "COUNTRY"
152
+ ],
153
+ "Relational": [
154
+ "ROLE",
155
+ "RELATIONSHIP"
156
+ ]
 
 
 
 
 
 
 
 
 
 
 
 
157
  }
158
+
159
+
160
 
161
 
162