Update app.py

app.py

@@ -16,13 +16,24 @@ nltk.download('stopwords')
 import matplotlib.pyplot as plt
 import numpy as np
 
-stopwords_list = stopwords.words('english') + ['your_additional_stopwords_here']
 
-# Define the model and tokenizer
-model_name = 'nlptown/bert-base-multilingual-uncased-sentiment'
-model = AutoModelForSequenceClassification.from_pretrained(model_name)
-tokenizer = AutoTokenizer.from_pretrained(model_name)
+stopwords_list = stopwords.words('english') + ['your_additional_stopwords_here']
 st.set_page_config(layout="wide")
+@st.cache_resource
+def load_model_and_tokenizer(model_name):
+    model = AutoModelForSequenceClassification.from_pretrained(model_name)
+    tokenizer = AutoTokenizer.from_pretrained(model_name)
+    return model, tokenizer
+
+model, tokenizer = load_model_and_tokenizer('nlptown/bert-base-multilingual-uncased-sentiment')
+
+@st.cache_resource
+def load_pipeline():
+    classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
+    return classifier
+
+classifier = load_pipeline()
+
 
 # Import the new model and tokenizer