Caleb Ellington
commited on
Commit
·
f9cb581
1
Parent(s):
4313dac
update with major refactor
Browse files
README.md
CHANGED
@@ -20,12 +20,12 @@ snapshot_download(repo_id=model_name, local_dir=genbio_models_path)
|
|
20 |
### Load model for inference
|
21 |
```python
|
22 |
import torch
|
23 |
-
from
|
24 |
|
25 |
ckpt_path = genbio_models_path.joinpath('model.ckpt')
|
26 |
model = SequenceRegression.load_from_checkpoint(ckpt_path, strict_loading=False).eval()
|
27 |
|
28 |
-
collated_batch = model.
|
29 |
logits = model(collated_batch)
|
30 |
print(logits)
|
31 |
```
|
|
|
20 |
### Load model for inference
|
21 |
```python
|
22 |
import torch
|
23 |
+
from modelgenerator.tasks import SequenceRegression
|
24 |
|
25 |
ckpt_path = genbio_models_path.joinpath('model.ckpt')
|
26 |
model = SequenceRegression.load_from_checkpoint(ckpt_path, strict_loading=False).eval()
|
27 |
|
28 |
+
collated_batch = model.transform({"sequences": ["ACGT", "AGCT"]})
|
29 |
logits = model(collated_batch)
|
30 |
print(logits)
|
31 |
```
|