dipsmom commited on
Commit
88951a9
·
verified ·
1 Parent(s): ea1bb28

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ image = Image.open("logo.webp") # Replace 'logo.webp' with your actual WEBP fil
13
  st.image(image, caption="Sentiment Analysis App", use_column_width=True)
14
 
15
  # Load the model and tokenizer
16
- @st.cache(allow_output_mutation=True)
17
  def load_model():
18
  tokenizer = BertTokenizer.from_pretrained("bert-base-uncased")
19
  model = BertForSequenceClassification.from_pretrained("bert-base-uncased", num_labels=8)
 
13
  st.image(image, caption="Sentiment Analysis App", use_column_width=True)
14
 
15
  # Load the model and tokenizer
16
+ @st.cache_resource
17
  def load_model():
18
  tokenizer = BertTokenizer.from_pretrained("bert-base-uncased")
19
  model = BertForSequenceClassification.from_pretrained("bert-base-uncased", num_labels=8)