Commit
·
bafeba8
1
Parent(s):
f701bda
Update README.md
Browse files
README.md
CHANGED
@@ -45,7 +45,7 @@ In addition to MentalT5, the MentaLLaMA project includes another model: MentaLLa
|
|
45 |
You can use the MentalT5 model in your Python project with the Hugging Face Transformers library. Here is a simple example of how to load the model:
|
46 |
|
47 |
```python
|
48 |
-
from transformers import T5Tokenizer, T5Model
|
49 |
tokenizer = T5Tokenizer.from_pretrained('Tianlin668/MentalT5')
|
50 |
model = T5Model.from_pretrained('Tianlin668/MentalT5')
|
51 |
```
|
|
|
45 |
You can use the MentalT5 model in your Python project with the Hugging Face Transformers library. Here is a simple example of how to load the model:
|
46 |
|
47 |
```python
|
48 |
+
from transformers import T5Tokenizer, T5Model
|
49 |
tokenizer = T5Tokenizer.from_pretrained('Tianlin668/MentalT5')
|
50 |
model = T5Model.from_pretrained('Tianlin668/MentalT5')
|
51 |
```
|