Downloads · 30 days
51
4% of all-time downloads
AI4Protein/ProPrime_650M_OGT_Prediction
ProPrime_650M_OGT_Prediction is a feature extraction model from AI4Protein. Use it when you need embeddings to search or compare text. It is set up for transformers.
Downloads · 30 days
51
4% of all-time downloads
All-time downloads
1.3K
Public
Parameters
669M
2.7 GB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors2.7 GB · 100%
From the Hugging Face model README
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)