Spaces:
Runtime error
Runtime error
magnolia-pm
commited on
Commit
•
ddae422
1
Parent(s):
80d1bd9
update
Browse files
app.py
CHANGED
@@ -68,15 +68,18 @@ input_text = st.text_input(
|
|
68 |
)
|
69 |
|
70 |
# desirability model
|
71 |
-
#
|
72 |
-
|
73 |
-
|
|
|
|
|
74 |
|
|
|
75 |
|
76 |
if 'tokenizer' not in globals():
|
77 |
tokenizer = AutoTokenizer.from_pretrained(
|
78 |
pretrained_model_name_or_path=model_path,
|
79 |
-
|
80 |
use_auth_token=auth_token
|
81 |
)
|
82 |
|
@@ -91,7 +94,7 @@ if 'model' not in globals():
|
|
91 |
# sentiment classifier
|
92 |
if 'classifier' not in globals():
|
93 |
sentiment_model = 'cardiffnlp/twitter-xlm-roberta-base-sentiment'
|
94 |
-
classifier = pipeline("sentiment-analysis", model=sentiment_model, tokenizer=sentiment_model, top_k=3)
|
95 |
|
96 |
classifier_output = classifier(input_text)
|
97 |
classifier_output_dict = {x['label']: x['score'] for x in classifier_output[0]}
|
|
|
68 |
)
|
69 |
|
70 |
# desirability model
|
71 |
+
# remote or local?
|
72 |
+
if os.environ.get("item-desirability"):
|
73 |
+
model_path = 'magnolia-psychometrics/item-desirability'
|
74 |
+
else:
|
75 |
+
model_path = '/nlp/nlp/models/finetuned/twitter-xlm-roberta-base-regressive-desirability-ft-4'
|
76 |
|
77 |
+
auth_token = os.environ.get("item-desirability") or True
|
78 |
|
79 |
if 'tokenizer' not in globals():
|
80 |
tokenizer = AutoTokenizer.from_pretrained(
|
81 |
pretrained_model_name_or_path=model_path,
|
82 |
+
use_fast=True,
|
83 |
use_auth_token=auth_token
|
84 |
)
|
85 |
|
|
|
94 |
# sentiment classifier
|
95 |
if 'classifier' not in globals():
|
96 |
sentiment_model = 'cardiffnlp/twitter-xlm-roberta-base-sentiment'
|
97 |
+
classifier = pipeline("sentiment-analysis", model=sentiment_model, tokenizer=sentiment_model, use_fast=False, top_k=3)
|
98 |
|
99 |
classifier_output = classifier(input_text)
|
100 |
classifier_output_dict = {x['label']: x['score'] for x in classifier_output[0]}
|