Update model_functions.py

model_functions.py

@@ -3,16 +3,18 @@ from transformers import (AutoModelForSequenceClassification, AutoModelForSeq2Se
                           AutoConfig, AutoModelForTokenClassification,
                           AutoTokenizer, pipeline)
 from peft import PeftModel, PeftConfig
+import streamlit as st
 
 
 
-
+@st.cache(show_spinner=False)
 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(show_spinner=False)
 def load_summarizer():
     config = PeftConfig.from_pretrained("marcelomoreno26/bart-large-samsum-adapter")
     model = AutoModelForSeq2SeqLM.from_pretrained("facebook/bart-large")
@@ -23,6 +25,7 @@ def load_summarizer():
 
     return tokenizer, model
 
+@st.cache(show_spinner=False)
 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)