shivam-1706 commited on
Commit
b25728f
·
verified ·
1 Parent(s): 51a45ae

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -9,7 +9,7 @@ import os
9
  import matplotlib.pyplot as plt
10
 
11
  # Load Models
12
- @st.cache_resource
13
  def load_models():
14
  bert_model = AutoModelForSequenceClassification.from_pretrained("shivam-1706/bert-model-stack")
15
  bert_tokenizer = AutoTokenizer.from_pretrained("shivam-1706/bert-model-stack")
 
9
  import matplotlib.pyplot as plt
10
 
11
  # Load Models
12
+ #@st.cache_resource
13
  def load_models():
14
  bert_model = AutoModelForSequenceClassification.from_pretrained("shivam-1706/bert-model-stack")
15
  bert_tokenizer = AutoTokenizer.from_pretrained("shivam-1706/bert-model-stack")