Peltarion
/

dnabert-distilbert

Inference Endpoints

Model card Files Files and versions Community

Joana Palés Huix commited on Jun 9, 2022

Commit

b117c15

•

1 Parent(s): 55342ab

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -17,7 +17,7 @@ The model can be used to fine-tune on a downstream genomic task, e.g. promoter i
 import torch
 from transformers import DistilBertForSequenceClassification
-model = DistilBertForSequenceClassification.from_pretrained(MODEL_NAME_OR_PATH)
 ```
 More details on how to fine-tune the model, dataset and additional source codes are available on [GitHub](https://github.com/joanaapa/Distillation-DNABERT-Promoter).

 import torch
 from transformers import DistilBertForSequenceClassification
+model = DistilBertForSequenceClassification.from_pretrained('Peltarion/dnabert-distilbert')
 ```
 More details on how to fine-tune the model, dataset and additional source codes are available on [GitHub](https://github.com/joanaapa/Distillation-DNABERT-Promoter).