IJAB commited on
Commit
86cc98f
·
1 Parent(s): 25bed32

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,10 +14,10 @@
14
  from transformers import pipeline
15
 
16
  # Load pre-trained NER model
17
- ner = pipeline('ner', model='bert-base-cased', tokenizer='bert-base-cased')
18
 
19
  # Define input text
20
- input_text = "The Mona Lisa is a 16th century portrait painted by Leonardo da Vinci."
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)