Spaces:
Sleeping
Sleeping
Update src/streamlit_app.py
Browse files- 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")
|