Spaces:
Runtime error
Runtime error
Tyler Burns
commited on
Commit
•
3952d5a
1
Parent(s):
341cb78
switched back to title for nltk keywords
Browse files- app.py +1 -1
- flycheck_app.py +1 -1
app.py
CHANGED
@@ -30,7 +30,7 @@ print("running sentence embeddings...")
|
|
30 |
# model_name = 'all-mpnet-base-v2'
|
31 |
model_name = 'all-MiniLM-L6-v2'
|
32 |
model = SentenceTransformer(model_name)
|
33 |
-
sentence_embeddings = model.encode(md['
|
34 |
sentence_embeddings = pd.DataFrame(sentence_embeddings)
|
35 |
|
36 |
# Reduce dimensionality
|
|
|
30 |
# model_name = 'all-mpnet-base-v2'
|
31 |
model_name = 'all-MiniLM-L6-v2'
|
32 |
model = SentenceTransformer(model_name)
|
33 |
+
sentence_embeddings = model.encode(md['title'].tolist(), show_progress_bar = True)
|
34 |
sentence_embeddings = pd.DataFrame(sentence_embeddings)
|
35 |
|
36 |
# Reduce dimensionality
|
flycheck_app.py
CHANGED
@@ -30,7 +30,7 @@ print("running sentence embeddings...")
|
|
30 |
# model_name = 'all-mpnet-base-v2'
|
31 |
model_name = 'all-MiniLM-L6-v2'
|
32 |
model = SentenceTransformer(model_name)
|
33 |
-
sentence_embeddings = model.encode(md['
|
34 |
sentence_embeddings = pd.DataFrame(sentence_embeddings)
|
35 |
|
36 |
# Reduce dimensionality
|
|
|
30 |
# model_name = 'all-mpnet-base-v2'
|
31 |
model_name = 'all-MiniLM-L6-v2'
|
32 |
model = SentenceTransformer(model_name)
|
33 |
+
sentence_embeddings = model.encode(md['title'].tolist(), show_progress_bar = True)
|
34 |
sentence_embeddings = pd.DataFrame(sentence_embeddings)
|
35 |
|
36 |
# Reduce dimensionality
|