Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -44,7 +44,7 @@ def text_to_sentiment(text):
44
 
45
  ##Company Extraction
46
  def fin_ner(text):
47
- api = gr.Interface.load("dslim/bert-base-NER", src='models', api_key=auth_token)
48
  replaced_spans = api(text)
49
  return replaced_spans
50
 
 
44
 
45
  ##Company Extraction
46
  def fin_ner(text):
47
+ api = gr.Interface.load("Jean-Baptiste/camembert-ner-with-dates", src='models', api_key=auth_token)
48
  replaced_spans = api(text)
49
  return replaced_spans
50