Update app.py
Browse filesStreamlit st.cache --> st.cache_resource
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.
|
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)
|