BulatF commited on
Commit
6b1b261
·
1 Parent(s): e94c73c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -22,7 +22,7 @@ from lime import lime_text
22
 
23
  stopwords_list = stopwords.words('english') + ['your_additional_stopwords_here']
24
  st.set_page_config(layout="wide")
25
- @st.cache_resource
26
  def load_model_and_tokenizer(model_name):
27
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
28
  tokenizer = AutoTokenizer.from_pretrained(model_name)
@@ -30,7 +30,7 @@ def load_model_and_tokenizer(model_name):
30
 
31
  model, tokenizer = load_model_and_tokenizer('nlptown/bert-base-multilingual-uncased-sentiment')
32
 
33
- @st.cache_resource
34
  def load_pipeline():
35
  classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
36
  return classifier
 
22
 
23
  stopwords_list = stopwords.words('english') + ['your_additional_stopwords_here']
24
  st.set_page_config(layout="wide")
25
+
26
  def load_model_and_tokenizer(model_name):
27
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
28
  tokenizer = AutoTokenizer.from_pretrained(model_name)
 
30
 
31
  model, tokenizer = load_model_and_tokenizer('nlptown/bert-base-multilingual-uncased-sentiment')
32
 
33
+
34
  def load_pipeline():
35
  classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
36
  return classifier