Upload README.md

README.md

@@ -26,7 +26,7 @@ pip3 install torch torchvision torchaudio --index-url https://download.pytorch.o
 pip install transformers
 pip install fair-esm
 ```
-Here is how to use TransHLA_I model to predict the peptide whether epitope:
+Here is how to use TransHLA_I model to predict whether a peptide is an epitope:
 
 ```python
 from transformers import AutoTokenizer
@@ -43,7 +43,7 @@ def pad_inner_lists_to_length(outer_list,target_length=16):
     return outer_list
 
 
-if __name__ = "__main__":
+if __name__ == "__main__":
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     print(f"Using {device} device")
     tokenizer = AutoTokenizer.from_pretrained("facebook/esm2_t33_650M_UR50D")