Update app.py

app.py

@@ -2,6 +2,10 @@ import streamlit as st
 from sentence_transformers.util import cos_sim
 from sentence_transformers import SentenceTransformer
 
+@st.cache
+def load_model():
+    model = SentenceTransformer('hackathon-pln-es/bertin-roberta-base-finetuning-esnli')
+    
 st.title("Sentence Embedding for Spanish with Bertin")
 st.text("Sentence embedding for spanish trained on NLI. Used for Sentence Textual Similarity.")
 
@@ -10,7 +14,7 @@ sent2 = st.text_area('Enter sentence 2')
 
 if st.button('Compute similarity'):
   if sent1 and sent2:
-    model = SentenceTransformer('hackathon-pln-es/bertin-roberta-base-finetuning-esnli')
+    model = load_model()
     encodings = model.encode([sent1, sent2])
     sim = cos_sim(encodings[0], encodings[1]).numpy().tolist()[0][0]
     st.text('Cosine Similarity: {0:.4f}'.format(sim))