Update README.md

README.md

@@ -19,13 +19,13 @@ from scipy.spatial.distance import cosine
 from transformers import AutoModel, AutoTokenizer
 
 # Load the model
-tokenizer = AutoTokenizer.from_pretrained("johngiorgi/declutr-base")
-model = AutoModel.from_pretrained("johngiorgi/declutr-base")
+tokenizer = AutoTokenizer.from_pretrained("johngiorgi/declutr-sci-base")
+model = AutoModel.from_pretrained("johngiorgi/declutr-sci-base")
 
 # Prepare some text to embed
 text = [
-    "A smiling costumed woman is holding an umbrella.",
-    "A happy woman in a fairy costume holds an umbrella.",
+    "Oncogenic KRAS mutations are common in cancer.",
+    "Notably, c-Raf has recently been found essential for development of K-Ras-driven NSCLCs.",
 ]
 inputs = tokenizer(text, padding=True, truncation=True, return_tensors="pt")