Update model_functions.py

model_functions.py

@@ -6,14 +6,14 @@ from peft import PeftModel, PeftConfig
 import streamlit as st
 
 
-
+@st.cache_resource
 def load_sentiment_analyzer():
     tokenizer = AutoTokenizer.from_pretrained("aliciiavs/sentiment-analysis-whatsapp2")
     model = AutoModelForSequenceClassification.from_pretrained("aliciiavs/sentiment-analysis-whatsapp2")
 
     return tokenizer, model
 
-
+@st.cache_resource
 def load_summarizer():
     config = PeftConfig.from_pretrained("marcelomoreno26/bart-large-samsum-adapter")
     model = AutoModelForSeq2SeqLM.from_pretrained("facebook/bart-large")
@@ -24,7 +24,7 @@ def load_summarizer():
 
     return tokenizer, model
 
-
+@st.cache_resource
 def load_NER():
     config = AutoConfig.from_pretrained("hannahisrael03/wikineural-multilingual-ner-finetuned-wikiann")
     model = AutoModelForTokenClassification.from_pretrained("hannahisrael03/wikineural-multilingual-ner-finetuned-wikiann",config=config)