pmikh26 commited on
Commit
847f637
·
verified ·
1 Parent(s): 6887942

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ from sentence_transformers import SentenceTransformer
10
  from sklearn.metrics.pairwise import cosine_similarity
11
 
12
 
13
- EMBED_MODEL_NAME = "sentence-transformers/all-MiniLM-L6-v2"
14
  CENTROIDS_PATH = "emotion_avg.pkl"
15
 
16
  with open(CENTROIDS_PATH, "rb") as f:
 
10
  from sklearn.metrics.pairwise import cosine_similarity
11
 
12
 
13
+ EMBED_MODEL_NAME = "sentence-transformers/all-mpnet-base-v2"
14
  CENTROIDS_PATH = "emotion_avg.pkl"
15
 
16
  with open(CENTROIDS_PATH, "rb") as f: