fondress commited on
Commit
612f05f
·
verified ·
1 Parent(s): 6829651

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -53,7 +53,7 @@ from transformers import AutoModel, AutoTokenizer
53
  # Load `PDeepPP` model
54
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
55
  print(f"Using {device} device")
56
- model = AutoModel.from_pretrained("YourModelName/`PDeepPP`", trust_remote_code=True)
57
  model.to(device)
58
 
59
  # Example protein sequences
@@ -77,7 +77,7 @@ from transformers import AutoModel
77
  # Load `PDeepPP` model
78
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
79
  print(f"Using {device} device")
80
- model = AutoModel.from_pretrained("YourModelName/`PDeepPP`", trust_remote_code=True)
81
  model.to(device)
82
 
83
  # Example protein sequences
 
53
  # Load `PDeepPP` model
54
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
55
  print(f"Using {device} device")
56
+ model = AutoModel.from_pretrained("fondress/PDeepPP_ACE", trust_remote_code=True)
57
  model.to(device)
58
 
59
  # Example protein sequences
 
77
  # Load `PDeepPP` model
78
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
79
  print(f"Using {device} device")
80
+ model = AutoModel.from_pretrained("fondress/PDeepPP_ACE", trust_remote_code=True)
81
  model.to(device)
82
 
83
  # Example protein sequences