Update app.py

app.py

@@ -1,11 +1,19 @@
 import gradio as gr
 from keras.layers import TextVectorization
 import tensorflow as tf
+import pandas as pd
+
+df = pd.read_csv('train.csv')
+
+X = df['comment_text']
+y = df[df.columns[2:]].values
 
 MAX_FEATURES = 200000
 vectorizer = TextVectorization(max_tokens=MAX_FEATURES,
                                output_sequence_length=1800,
                                output_mode='int')
+vectorizer.adapt(X.values)
+
 model = tf.keras.models.load_model('pfilter.h5')
 
 
@@ -22,5 +30,5 @@ def score_comment(comment):
 
 interface = gr.Interface(fn=score_comment,
                          inputs=gr.components.Textbox
-                         (lines=2, placeholder='Comment to Score'),outputs = 'text')
+                         (lines=2, placeholder='Comment to Score'), outputs='text')
 interface.launch(inline=False)