Caleb Ellington commited on
Commit
f9cb581
·
1 Parent(s): 4313dac

update with major refactor

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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 genbio_finetune.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.collate({"sequences": ["ACGT", "AGCT"]})
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
  ```