akdeniz27 commited on
Commit
5578878
1 Parent(s): 73d0329

Update app.py

Browse files

Streamlit st.cache --> st.cache_resource

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -62,7 +62,7 @@ elif labels == "Label List #2": selected_labels = label_list_2
62
  elif labels == "New Label List":
63
  selected_labels = st.text_area("New Label List (Pls Input as comma-separated)", value="", height=16).split(",")
64
 
65
- @st.cache(allow_output_mutation=True)
66
  def setModel(model_checkpoint):
67
  model = AutoModelForSequenceClassification.from_pretrained(model_checkpoint)
68
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
62
  elif labels == "New Label List":
63
  selected_labels = st.text_area("New Label List (Pls Input as comma-separated)", value="", height=16).split(",")
64
 
65
+ @st.cache_resource
66
  def setModel(model_checkpoint):
67
  model = AutoModelForSequenceClassification.from_pretrained(model_checkpoint)
68
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)