Update pages/Named Entity Recognition.py
Browse files
pages/Named Entity Recognition.py
CHANGED
|
@@ -231,7 +231,8 @@ ner_mapping = {
|
|
| 231 |
'DRV': 'Driver'
|
| 232 |
}
|
| 233 |
|
| 234 |
-
|
|
|
|
| 235 |
model = st.sidebar.selectbox("Choose the pretrained model", model_list, help="For more info about the models visit: https://sparknlp.org/models")
|
| 236 |
st.title(language_info[abbreviation_map[langauge]]["title"])
|
| 237 |
|
|
|
|
| 231 |
'DRV': 'Driver'
|
| 232 |
}
|
| 233 |
|
| 234 |
+
langauge = 'English'
|
| 235 |
+
model_list = language_info[abbreviation_map[langauge]]["model_name"]
|
| 236 |
model = st.sidebar.selectbox("Choose the pretrained model", model_list, help="For more info about the models visit: https://sparknlp.org/models")
|
| 237 |
st.title(language_info[abbreviation_map[langauge]]["title"])
|
| 238 |
|