Update README.md
Browse files
README.md
CHANGED
|
@@ -111,8 +111,8 @@ def mean_pooling(model_output, attention_mask):
|
|
| 111 |
sentences = ['This is an example sentence', 'Each sentence is converted']
|
| 112 |
|
| 113 |
# Load model from HuggingFace Hub
|
| 114 |
-
tokenizer = AutoTokenizer.from_pretrained('
|
| 115 |
-
model = AutoModel.from_pretrained('
|
| 116 |
|
| 117 |
# Tokenize sentences
|
| 118 |
encoded_input = tokenizer(sentences, padding=True, truncation=True, return_tensors='pt')
|
|
|
|
| 111 |
sentences = ['This is an example sentence', 'Each sentence is converted']
|
| 112 |
|
| 113 |
# Load model from HuggingFace Hub
|
| 114 |
+
tokenizer = AutoTokenizer.from_pretrained('shtilev/medical_embedded_v1')
|
| 115 |
+
model = AutoModel.from_pretrained('shtilev/medical_embedded_v1')
|
| 116 |
|
| 117 |
# Tokenize sentences
|
| 118 |
encoded_input = tokenizer(sentences, padding=True, truncation=True, return_tensors='pt')
|