Update app.py

app.py

@@ -153,8 +153,12 @@ tokenizer = transformers.AutoTokenizer.from_pretrained("DeeeTeeee01/mytest_train
 def tokenizer_hash_func(tokenizer):
     return tokenizer.__str__()
 
+# Define the custom hash function for tokenizers.AddedToken
+def added_token_hash_func(token):
+    return token.__str__()
+
 # Define the function for sentiment analysis
-@st.cache(hash_funcs={tokenizers.Tokenizer: tokenizer_hash_func})  # Specify the custom hash function
+@st.cache(hash_funcs={tokenizers.Tokenizer: tokenizer_hash_func, tokenizers.AddedToken: added_token_hash_func})
 def predict_sentiment(text):
     # Load the pipeline
     pipeline = transformers.pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
@@ -166,6 +170,7 @@ def predict_sentiment(text):
 
     return sentiment, score
 
+
 # Setting the page configurations
 st.set_page_config(
     page_title="Sentiment Analysis App",