Modelo PyTorch MLP para clasificación de familias de proteínas Pfam.
1from huggingface_hub import hf_hub_download
2import torch, pickle
3
4model_path = hf_hub_download(repo_id="Fb56816/protein-classifier-model", filename="pfam_model_5513.pt")
5scaler_path = hf_hub_download(repo_id="Fb56816/protein-classifier-model", filename="pfam_scaler.pkl")
6encoder_path = hf_hub_download(repo_id="Fb56816/protein-classifier-model", filename="pfam_encoder.pkl")
7
8model = torch.load(model_path, map_location="cpu", weights_only=False)
9scaler = pickle.load(open(scaler_path, "rb"))
10encoder = pickle.load(open(encoder_path, "rb"))