Update app.py
Browse files
app.py
CHANGED
|
@@ -5,6 +5,7 @@ from sentence_transformers import SentenceTransformer
|
|
| 5 |
@st.cache
|
| 6 |
def load_model():
|
| 7 |
model = SentenceTransformer('hackathon-pln-es/bertin-roberta-base-finetuning-esnli')
|
|
|
|
| 8 |
return model
|
| 9 |
|
| 10 |
st.title("Sentence Embedding for Spanish with Bertin")
|
|
|
|
| 5 |
@st.cache
|
| 6 |
def load_model():
|
| 7 |
model = SentenceTransformer('hackathon-pln-es/bertin-roberta-base-finetuning-esnli')
|
| 8 |
+
model.eval()
|
| 9 |
return model
|
| 10 |
|
| 11 |
st.title("Sentence Embedding for Spanish with Bertin")
|