rajistics commited on
Commit
8fdf4b9
1 Parent(s): c261897

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -56,7 +56,7 @@ def fin_ext(text):
56
  ##Forward Looking Statement
57
  def fls(text):
58
  # fls_model = pipeline("text-classification", model="yiyanghkust/finbert-fls", tokenizer="yiyanghkust/finbert-fls")
59
- fls_model = pipeline("text-classification", model="demo-org/finbert_fls", tokenizer="demo-org/finbert_fls", api_key=auth_token)
60
  results = fls_model(split_in_sentences(text))
61
  return make_spans(text,results)
62
 
56
  ##Forward Looking Statement
57
  def fls(text):
58
  # fls_model = pipeline("text-classification", model="yiyanghkust/finbert-fls", tokenizer="yiyanghkust/finbert-fls")
59
+ fls_model = pipeline("text-classification", model="demo-org/finbert_fls", tokenizer="demo-org/finbert_fls", use_auth_token=auth_token)
60
  results = fls_model(split_in_sentences(text))
61
  return make_spans(text,results)
62