Update app.py

app.py

@@ -15,6 +15,7 @@ else:
 
 tweets_raw = pd.read_csv("train.csv", nrows=20)
 
+label_set = ["toxic", "severe_toxic", "obscene", "threat", "insult", "identity_hate"]
 
 # Run encoding through model to get classification output.
 encoding = tokenizer.encode(txt, return_tensors='pt')