Hugging Face's logo

jiaxie
/
DeepProtT5-SubCellular

Safetensors
t5
Model card Files
xet
Community
jiaxie commited on
Commit
90d4a56
·
verified ·
1 Parent(s): 80192fe

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -50,6 +50,6 @@ class T5ClassificationModel(PreTrainedModel):
50
  ```
51
  Then we load our pretrained model
52
  ```python
53
- tokenizer = T5Tokenizer.from_pretrained("jiaxie/DeepProtT5-Solubility", do_lower_case=False)
54
- model = T5ClassificationModel.from_pretrained("jiaxie/DeepProtT5-Solubility", torch_dtype=torch.bfloat16).to("cuda")
55
  ```
 
50
  ```
51
  Then we load our pretrained model
52
  ```python
53
+ tokenizer = T5Tokenizer.from_pretrained("jiaxie/DeepProtT5-SubCellular", do_lower_case=False)
54
+ model = T5ClassificationModel.from_pretrained("jiaxie/DeepProtT5-SubCellular", torch_dtype=torch.bfloat16).to("cuda")
55
  ```