Elvan Selvano commited on
Commit
565d6fb
β€’
1 Parent(s): bb46736

Use static folder

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ import torch
8
 
9
  @st.cache(allow_output_mutation=True)
10
  def load_model():
11
- return SentenceTransformer('all-MiniLM-L6-v2')
12
 
13
  def find_top_similar(sentence, corpus_sentences, corpus_embeddings):
14
 
 
8
 
9
  @st.cache(allow_output_mutation=True)
10
  def load_model():
11
+ return SentenceTransformer('minilm_sbert')
12
 
13
  def find_top_similar(sentence, corpus_sentences, corpus_embeddings):
14