Spaces:

JohnKouf
/

transformer_api

Runtime error

JohnKouf commited on Jun 26

Commit

2d7f83d

verified ·

1 Parent(s): cbd145d

Update Similarity.py

Files changed (1) hide show

Similarity.py CHANGED Viewed

@@ -7,7 +7,7 @@ class Similarity:
     def load_model(self):
         if self.model is None:
             print("Loading SentenceTransformer model...")
-            self.model = SentenceTransformer("lighteternal/stsb-xlm-r-greek-transfer")
             print("Model loaded.")
     def embed_text(self, text):

     def load_model(self):
         if self.model is None:
             print("Loading SentenceTransformer model...")
+            self.model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
             print("Model loaded.")
     def embed_text(self, text):