Spaces:

cools
/

Gideon

Runtime error

cools commited on Jul 14, 2023

Commit

d1efa43

1 Parent(s): 19729fa

Larger sbert model

Files changed (1) hide show

Tagger.py CHANGED Viewed

@@ -17,7 +17,7 @@ def get_paragraphed_text(folderpath):
 def semantic_match(template, corpus):
-    model = SentenceTransformer('paraphrase-MiniLM-L6-v2')
     embs = model.encode(corpus)
     if type(template) == list:
         template_emb = model.encode(template)

 def semantic_match(template, corpus):
+    model = SentenceTransformer('all-mpnet-base-v2')
     embs = model.encode(corpus)
     if type(template) == list:
         template_emb = model.encode(template)