Update app.py

app.py

@@ -14,10 +14,14 @@ def load_model(filepath):
     return tf.keras.models.load_model(filepath)
 
 # Load tokenizer
+import pickle
+
+# Function to load the tokenizer
 @st.cache_resource
-def load_tokenizer(filepath):
-    with open(filepath, 'rb') as handle:
-        return pickle.load(handle)
+def load_tokenizer():
+    with open('tokenizer.pickle', 'rb') as handle:
+        tokenizer = pickle.load(handle)
+    return tokenizer
 
 # Preprocess prompt
 def preprocess_prompt(prompt, tokenizer, max_length=100):