Hugging Face's logo

Spaces:
CATIE-AQ
/
NERmembert
Running

App Files Community
bourdoiscatie commited on
Commit
dc8f7ad
·
verified ·
1 Parent(s): 82b8822

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ def load_model(version,text):
32
  ner = pipeline('token-classification', model='CATIE-AQ/NERmembert-large-3entities', tokenizer='CATIE-AQ/NERmembert-large-3entities', aggregation_strategy="simple")
33
  result = ner(text)
34
  return result
35
- if version == "NERmemBERT1-4entities":
36
  ner = pipeline('token-classification', model='CATIE-AQ/NERmembert-base-4entities', tokenizer='CATIE-AQ/NERmembert-base-4entities', aggregation_strategy="simple")
37
  result = ner(text)
38
  return result
 
32
  ner = pipeline('token-classification', model='CATIE-AQ/NERmembert-large-3entities', tokenizer='CATIE-AQ/NERmembert-large-3entities', aggregation_strategy="simple")
33
  result = ner(text)
34
  return result
35
+ if version == "NERmemBERT1-4entities-base":
36
  ner = pipeline('token-classification', model='CATIE-AQ/NERmembert-base-4entities', tokenizer='CATIE-AQ/NERmembert-base-4entities', aggregation_strategy="simple")
37
  result = ner(text)
38
  return result