Ariel Hsieh commited on
Commit
d8d9ab4
1 Parent(s): ef71cfe
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ st.title("Toxic Tweets")
21
  tokenizer = AutoTokenizer.from_pretrained("Ariel8/toxic-tweets-classification")
22
  model = AutoModelForSequenceClassification.from_pretrained("Ariel8/toxic-tweets-classification")
23
 
24
- X_train = ["Why is Owen's retirement from football not mentioned? He hasn't played a game since 2005."]
25
  batch = tokenizer(X_train, truncation=True, padding='max_length', return_tensors="pt")
26
  labels = ["toxic", "severe_toxic", "obscene", "threat", "insult", "identity_hate"]
27
 
 
21
  tokenizer = AutoTokenizer.from_pretrained("Ariel8/toxic-tweets-classification")
22
  model = AutoModelForSequenceClassification.from_pretrained("Ariel8/toxic-tweets-classification")
23
 
24
+ X_train = ["BlackKite being a faggot"]
25
  batch = tokenizer(X_train, truncation=True, padding='max_length', return_tensors="pt")
26
  labels = ["toxic", "severe_toxic", "obscene", "threat", "insult", "identity_hate"]
27