Update app.py

app.py

@@ -3,6 +3,9 @@ import numpy as np
 from transformers import BertTokenizer, BertForSequenceClassification
 import torch
 
+# Add title
+st.title("Toxic Comment Classification")
+
 @st.cache_resource
 def get_model():
     tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')