kusa04 commited on
Commit
763f466
·
verified ·
1 Parent(s): 1d43e3c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=-1) # -1 to 0
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: