Update README.md
Browse files
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("
|
| 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("
|
| 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
|