Rob Caamano
commited on
Update app.py
Browse files
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 |
-
"
|
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")
|