mstaron
/

SingBERTa

mstaron commited on Feb 7, 2023

Commit

c389e88

•

1 Parent(s): da26daa

Update README.md

Files changed (1) hide show

README.md CHANGED Viewed

@@ -33,8 +33,8 @@ from transformers import pipeline
 # the models are already pre-defined, so we do not need to train anything here
 features = pipeline(
     "feature-extraction",
-    model="./SingletonSSLBERT",
-    tokenizer="./SingletonSSLBERT",
     return_tensor = False
 )

 # the models are already pre-defined, so we do not need to train anything here
 features = pipeline(
     "feature-extraction",
+    model=model,
+    tokenizer=tokenizer,
     return_tensor = False
 )