Update app.py

app.py

@@ -21,7 +21,7 @@ def embedding_and_index():
 
 embeddings_array, index = embedding_and_index()
      
-@st.cache(allow_output_mutation=True)
+@st.cache_resource
 def load_tokenizer_and_model():
     tokenizer = AutoTokenizer.from_pretrained("DeepPavlov/rubert-base-cased-sentence")
     model = AutoModel.from_pretrained("DeepPavlov/rubert-base-cased-sentence")
@@ -29,7 +29,7 @@ def load_tokenizer_and_model():
     
 tokenizer, model = load_tokenizer_and_model()
 
-@st.cache(allow_output_mutation=True)
+@st.cache_resource
 def encode_description(description, tokenizer, model):
     tokens = tokenizer(description, return_tensors="pt")
     with torch.no_grad():