Update README.md
Browse files
README.md
CHANGED
|
@@ -154,6 +154,7 @@ tokenizer = AutoTokenizer.from_pretrained(cfg.pretrained_model_name_or_path)
|
|
| 154 |
model = ProtoForMultiLabelClassification.from_pretrained(
|
| 155 |
"row56/ProtoPatient",
|
| 156 |
config=cfg,
|
|
|
|
| 157 |
)
|
| 158 |
model.to(device)
|
| 159 |
model.eval()
|
|
|
|
| 154 |
model = ProtoForMultiLabelClassification.from_pretrained(
|
| 155 |
"row56/ProtoPatient",
|
| 156 |
config=cfg,
|
| 157 |
+
|
| 158 |
)
|
| 159 |
model.to(device)
|
| 160 |
model.eval()
|