Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -33,7 +33,7 @@ def load_sentiment_pipeline(): # sentiment pipeline
|
|
33 |
"cardiffnlp/twitter-roberta-base-sentiment-latest",
|
34 |
use_auth_token=st.secrets["hugging_face_token"]
|
35 |
)
|
36 |
-
sentiment_pipeline = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer, device
|
37 |
max_tokens = tokenizer.model_max_length
|
38 |
|
39 |
if max_tokens > 10000:
|
|
|
33 |
"cardiffnlp/twitter-roberta-base-sentiment-latest",
|
34 |
use_auth_token=st.secrets["hugging_face_token"]
|
35 |
)
|
36 |
+
sentiment_pipeline = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer, device=0) # -1 to 0
|
37 |
max_tokens = tokenizer.model_max_length
|
38 |
|
39 |
if max_tokens > 10000:
|