Tumo505
/

SSL-ECG-Classification-model-card

self-supervised-learning

Eval Results (legacy)

Model card Files Files and versions

Tumo505 commited on Apr 19

Commit

ddc5b19

·

1 Parent(s): 6da40bb

update model loading code

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -79,7 +79,7 @@ import torch
 from transformers import AutoModel
 # Load model
-model = AutoModel.from_pretrained("Tumo505/ssl-ecg-simclr-finetuned", trust_remote_code=True)
 model.eval()
 # Prepare 12-lead ECG (batch_size, 12 leads, 5000 samples)

 from transformers import AutoModel
 # Load model
+model = AutoModel.from_pretrained("Tumo505/SSL-ECG-Classificcation", trust_remote_code=True)
 model.eval()
 # Prepare 12-lead ECG (batch_size, 12 leads, 5000 samples)