rajistics commited on
Commit
b628666
1 Parent(s): edab7cf

adding private model

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -55,7 +55,8 @@ def fin_ext(text):
55
 
56
  ##Forward Looking Statement
57
  def fls(text):
58
- fls_model = pipeline("text-classification", model="yiyanghkust/finbert-fls", tokenizer="yiyanghkust/finbert-fls")
 
59
  results = fls_model(split_in_sentences(text))
60
  return make_spans(text,results)
61
 
 
55
 
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_flsyiyanghkust/finbert-fls", api_key=auth_token)
60
  results = fls_model(split_in_sentences(text))
61
  return make_spans(text,results)
62