mila-intel
/

ProtST-esm1b

Feature Extraction

Model card Files Files and versions Community

Jiqing commited on Apr 2

Commit

91a0e26

•

1 Parent(s): 921086f

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -106,7 +106,7 @@ if __name__ == "__main__":
             desc="Running tokenize_proteins on dataset",
         )
-    labels = load_dataset("Jiqing/subloc_template", cache_dir="~/.cache/huggingface/datasets")["train"]["name"]
     text_tokenizer.encode(labels[0], max_length=128, truncation=True, add_special_tokens=False)
     label_feature = label_embedding(labels, text_tokenizer, text_model, device)

             desc="Running tokenize_proteins on dataset",
         )
+    labels = load_dataset("mila-intel/subloc_template", cache_dir="~/.cache/huggingface/datasets")["train"]["name"]
     text_tokenizer.encode(labels[0], max_length=128, truncation=True, add_special_tokens=False)
     label_feature = label_embedding(labels, text_tokenizer, text_model, device)