ProPrime

from transformers import AutoModel, AutoTokenizer
model_path = "AI4Protein/ProPrime_650M_OGT_Prediction"

model = AutoModel.from_pretrained(model_path, trust_remote_code=True)
tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)

seqs = [
    "ADHJSJHSLGJSGLSGGAD",
    "MSJKFHLSKGJSG",
    "SDKHF"
]

inputs = tokenizer(seqs, padding=True, return_tensors="pt")
outputs = model(**inputs)
print(outputs.predicted_values)