Hugging Face's logo

Spaces:
CATIE-AQ
/
NERmembert
Running

App Files Community
bourdoiscatie commited on
Commit
26f44f6
·
verified ·
1 Parent(s): 982ebb8

Update app.py

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