Tyler Burns commited on
Commit
0df1ea2
1 Parent(s): 7aa8bfe

added smaller model for the sake of speed

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -20,7 +20,8 @@ md = pd.DataFrame(md)
20
 
21
  # Load the model
22
  print("running sentence embeddings...")
23
- model_name = 'all-mpnet-base-v2'
 
24
  model = SentenceTransformer(model_name)
25
  sentence_embeddings = model.encode(md['body'].tolist(), show_progress_bar = True)
26
  sentence_embeddings = pd.DataFrame(sentence_embeddings)
 
20
 
21
  # Load the model
22
  print("running sentence embeddings...")
23
+ # model_name = 'all-mpnet-base-v2'
24
+ model_name = 'all-MiniLM-L6-v2'
25
  model = SentenceTransformer(model_name)
26
  sentence_embeddings = model.encode(md['body'].tolist(), show_progress_bar = True)
27
  sentence_embeddings = pd.DataFrame(sentence_embeddings)