Hugging Face's logo

Spaces:
CATIE-AQ
/
NERmembert
Running

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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ def load_model(version,text):
40
  ner = pipeline('token-classification', model='CATIE-AQ/NERmembert2-4entities', tokenizer='CATIE-AQ/NERmembert2-4entities', aggregation_strategy="simple")
41
  result = ner(text)
42
  return result
43
- if version == "NERmemBERTa-4entities-base":
44
  ner = pipeline('token-classification', model='CATIE-AQ/NERmemberta-4entities', tokenizer='CATIE-AQ/NERmemberta-4entities', aggregation_strategy="simple")
45
  result = ner(text)
46
  return result
 
40
  ner = pipeline('token-classification', model='CATIE-AQ/NERmembert2-4entities', tokenizer='CATIE-AQ/NERmembert2-4entities', aggregation_strategy="simple")
41
  result = ner(text)
42
  return result
43
+ if version == "NERmemBERTa-4entities":
44
  ner = pipeline('token-classification', model='CATIE-AQ/NERmemberta-4entities', tokenizer='CATIE-AQ/NERmemberta-4entities', aggregation_strategy="simple")
45
  result = ner(text)
46
  return result