Rename model

app.py

@@ -8,7 +8,7 @@ st.set_page_config(layout='wide')
 @st.cache(allow_output_mutation=True)
 def load_model():
     """Load pretrained model from SentenceTransformer"""
-    return SentenceTransformer('all-MiniLM-L6-v2')
+    return SentenceTransformer('minilm_sbert')
 
 def semantic_search(model, sentence, corpus_embeddings):
     """Perform semantic search on the corpus"""
@@ -94,15 +94,6 @@ def main():
         st.info(f'Showing results for {job_title}')
         result = get_similarity_score(model, data, job_title, corpus_embeddings)
         result = result[columns]
-
-        # st.download_button(
-        #     "Download Table",
-        #     result.to_csv().encode('utf-8'),
-        #     "result.csv",
-        #     "text/csv",
-        #     key='download-csv'
-        # )
-
         show_aggrid_table(result)
 
 if __name__ == '__main__':