MishaD
commited on
Commit
·
9d1e5f8
1
Parent(s):
55dd195
Testing change st.cache_data to st.cache
Browse files
model.py
CHANGED
|
@@ -34,7 +34,7 @@ class SimpleClassifier(nn.Module):
|
|
| 34 |
return X
|
| 35 |
|
| 36 |
|
| 37 |
-
@st.
|
| 38 |
def load_models():
|
| 39 |
model = RobertaForMaskedLM.from_pretrained("roberta-base")
|
| 40 |
model.lm_head = nn.Identity()
|
|
|
|
| 34 |
return X
|
| 35 |
|
| 36 |
|
| 37 |
+
@st.cache()
|
| 38 |
def load_models():
|
| 39 |
model = RobertaForMaskedLM.from_pretrained("roberta-base")
|
| 40 |
model.lm_head = nn.Identity()
|