update with major refactor

README.md

@@ -11,11 +11,11 @@ snapshot_download(repo_id=model_name, local_dir=genbio_models_path)
 ```
 ### Load model for inference
 ```python
-from genbio_finetune.tasks import SequenceRegression
+from modelgenerator.tasks import SequenceRegression
 
 ckpt_path = genbio_models_path.joinpath('fold0', 'model.ckpt')
 model = SequenceRegression.load_from_checkpoint(ckpt_path, strict_loading=False).eval()
 
-collated_batch = model.collate({"sequences": ["ACGT", "AGCT"]})
+collated_batch = model.transform({"sequences": ["ACGT", "AGCT"]})
 logits = model(collated_batch)
 print(logits)