Update app.py

app.py

@@ -66,6 +66,7 @@ else:
 
     ## Load the model
     import transformers
+    from functools import lru_cache
 
     import torch
     import torch.nn.functional as F
@@ -80,7 +81,8 @@ else:
 
     model, tokenizer = load_model_and_tokenizer()
 
-    @st.cache_data
+    @st.cache_resource
+    @lru_cache(maxsize=128)
     def encode_sentences(sent0, sent1):
         sentences = [sent0, sent1]
         tokens = []