Update app.py

app.py

@@ -1,8 +1,21 @@
 import gradio as gr
 import tensorflow as tf
+import pandas as pd
+from tensorflow.keras.layers import TextVectorization
 
 model = tf.keras.models.load_model('toxic-detect.h5')
 
+df = pd.read_csv('train.csv')
+
+X = df.comment_text
+
+vectorizer = TextVectorization(max_tokens=200000, 
+                               output_sequence_length=1800, 
+                               output_mode='int')
+
+vectorizer.adapt(X.values)
+
+
 def evaluate_comment(Comment):
     processed_Comment = vectorizer([Comment])
     res = model.predict(processed_Comment)