Update toxic.py

toxic.py

@@ -2,9 +2,14 @@ import streamlit as st
 import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 
+
+def app():
+    st.title('Toxic Comment Classifier')
+    st.write('This is the toxic comment classifier page.')
 model_checkpoint = 'cointegrated/rubert-tiny-toxicity'
 tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
 model = AutoModelForSequenceClassification.from_pretrained(model_checkpoint)
+
 if torch.cuda.is_available():
     model.cuda()