Update README.md
Browse files
README.md
CHANGED
|
@@ -79,7 +79,7 @@ def load_and_prepare_conversations(filepath):
|
|
| 79 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
| 80 |
import torch
|
| 81 |
|
| 82 |
-
model_name = "
|
| 83 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 84 |
model = AutoModelForSequenceClassification.from_pretrained(model_name)
|
| 85 |
model.eval().to("cuda" if torch.cuda.is_available() else "cpu")
|
|
|
|
| 79 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
| 80 |
import torch
|
| 81 |
|
| 82 |
+
model_name = "webesama/MADRS-BERT"
|
| 83 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 84 |
model = AutoModelForSequenceClassification.from_pretrained(model_name)
|
| 85 |
model.eval().to("cuda" if torch.cuda.is_available() else "cpu")
|