Rob Caamano commited on
Commit
d8d3fab
·
unverified ·
1 Parent(s): e2d0da3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ model_name = "cardiffnlp/twitter-roberta-base-sentiment"
16
  tokenizer = AutoTokenizer.from_pretrained(model_name)
17
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
18
  clf = pipeline(
19
- "toxicity-analysis", model=model, tokenizer=tokenizer, return_all_scores=True
20
  )
21
 
22
  input = tokenizer(text, return_tensors="tf")
 
16
  tokenizer = AutoTokenizer.from_pretrained(model_name)
17
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
18
  clf = pipeline(
19
+ "sentiment-analysis", model=model, tokenizer=tokenizer, return_all_scores=True
20
  )
21
 
22
  input = tokenizer(text, return_tensors="tf")