Update README.md
Browse files
README.md
CHANGED
|
@@ -56,8 +56,8 @@ You can load the model using Hugging Face's `transformers` library:
|
|
| 56 |
```python
|
| 57 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 58 |
|
| 59 |
-
tokenizer = AutoTokenizer.from_pretrained("SriyaM/
|
| 60 |
-
model = AutoModelForCausalLM.from_pretrained("SriyaM/
|
| 61 |
|
| 62 |
prompt = "Write a motivational message for someone in the contemplation stage thinking about starting to exercise."
|
| 63 |
inputs = tokenizer(prompt, return_tensors="pt")
|
|
|
|
| 56 |
```python
|
| 57 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 58 |
|
| 59 |
+
tokenizer = AutoTokenizer.from_pretrained("SriyaM/MHC-Coach")
|
| 60 |
+
model = AutoModelForCausalLM.from_pretrained("SriyaM/MHC-Coach")
|
| 61 |
|
| 62 |
prompt = "Write a motivational message for someone in the contemplation stage thinking about starting to exercise."
|
| 63 |
inputs = tokenizer(prompt, return_tensors="pt")
|