Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import pickle
 
 import tensorflow as tf
 from tensorflow import keras
@@ -17,6 +18,11 @@ EMBED_DIM = 512
 # Per-layer units in the feed-forward network
 FF_DIM = 512
 
+# load the text data
+open_file = open('text.pkl', "rb")
+text_data = pickle.load(open_file)
+open_file.close()
+
 # text preprocessing
 def custom_standardization(input_string):
     lowercase = tf.strings.lower(input_string)