nppmatt commited on
Commit
fd91de6
1 Parent(s): 2db5d22

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -17,8 +17,8 @@ bert_path = "bert-base-uncased"
17
  tokenizer = AutoTokenizer.from_pretrained(bert_path)
18
 
19
  # Read and format data.
20
- tweets_raw = pd.read_csv("test.csv", nrows=50)
21
- labels_raw = pd.read_csv("test_labels.csv", nrows=50)
22
 
23
  label_set = ["toxic", "severe_toxic", "obscene", "threat", "insult", "identity_hate"]
24
  label_vector = labels_raw[label_set].values.tolist()
 
17
  tokenizer = AutoTokenizer.from_pretrained(bert_path)
18
 
19
  # Read and format data.
20
+ tweets_raw = pd.read_csv("test.csv", nrows=200)
21
+ labels_raw = pd.read_csv("test_labels.csv", nrows=200)
22
 
23
  label_set = ["toxic", "severe_toxic", "obscene", "threat", "insult", "identity_hate"]
24
  label_vector = labels_raw[label_set].values.tolist()