Spaces:

JohnKouf
/

transformer_api

Runtime error

JohnKouf commited on Jun 26

Commit

80906c5

verified ·

1 Parent(s): 04434ba

Update Similarity.py

Files changed (1) hide show

Similarity.py CHANGED Viewed

@@ -2,8 +2,8 @@ import nltk
 from sentence_transformers import util
 class Similarity:
-    def __init__(self, model):
-        self.model = model
         # Make sure nltk punkt tokenizer is downloaded
         nltk.download('punkt')

 from sentence_transformers import util
 class Similarity:
+    def __init__(self):
+        self.model = SentenceTransformer("lighteternal/stsb-xlm-r-greek-transfer")
         # Make sure nltk punkt tokenizer is downloaded
         nltk.download('punkt')