Update app.py

app.py

@@ -8,16 +8,14 @@ from tensorflow import keras
 
 model = tf.keras.models.load_model('toxicity_model.h5')
 
-df = pd.read_csv('train.csv')
-X = df['comment_text']
-y = df[df.columns[2:]].values
+dataset = pd.read_csv('train.csv')
+comments = dataset['comment_text']
 
-MAX_FEATURES = 200000
-vectorizer = TextVectorization(max_tokens=MAX_FEATURES,
+vectorizer = TextVectorization(max_tokens = 2500000,
                                output_sequence_length=1800,
                                output_mode='int')
 
-vectorizer.adapt(X.values)
+vectorizer.adapt(comments.values)
 
 st.title('Toxicity Classifier')
 
@@ -52,10 +50,7 @@ if(identity_hate > highest_class_rating):
     highest_class_rating = identity_hate
 
 st.write("---")
-st.write("Toxicity rating:")
-st.write(toxicity)
+st.write("Overall toxicity rating: " +toxicity)
 st.write("---")
-st.write("Highest Toxicity Class:")
-st.write(highest_class)
-st.write("Rating:")
-st.write(highest_class_rating)
+st.write("Highest toxicity class:" +highest_class)
+st.write("Rating:" +highest_class_rating)