Spaces:
Runtime error
Runtime error
Tyler Burns
commited on
Commit
•
0df1ea2
1
Parent(s):
7aa8bfe
added smaller model for the sake of speed
Browse files
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)
|