adding private model
Browse files
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 |
|