Update app.py
Browse files
app.py
CHANGED
@@ -14,10 +14,10 @@
|
|
14 |
from transformers import pipeline
|
15 |
|
16 |
# Load pre-trained NER model
|
17 |
-
ner = pipeline('ner', model='
|
18 |
|
19 |
# Define input text
|
20 |
-
input_text = "
|
21 |
|
22 |
# Call the NER pipeline on the input text
|
23 |
ner_results = ner(input_text)
|
|
|
14 |
from transformers import pipeline
|
15 |
|
16 |
# Load pre-trained NER model
|
17 |
+
ner = pipeline('ner', model='has-abi/distilBERT-finetuned-resumes-sections', tokenizer='bert-base-cased')
|
18 |
|
19 |
# Define input text
|
20 |
+
input_text = st.text_area("Enter text: ")
|
21 |
|
22 |
# Call the NER pipeline on the input text
|
23 |
ner_results = ner(input_text)
|