bputzeys commited on
Commit
85d44c8
·
verified ·
1 Parent(s): 2067fe9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -37,7 +37,7 @@ pip install --upgrade helical
37
 
38
  #### Generate Embeddings
39
  ```python
40
- from helical import HelixmRNA, HelixmRNAConfig
41
  import torch
42
 
43
  device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -57,7 +57,7 @@ embeddings = helix_mrna.get_embeddings(processed_input_data)
57
  #### Fine-Tuning
58
  Classification fine-tuning example:
59
  ```python
60
- from helical import HelixmRNAFineTuningModel, HelixmRNAConfig
61
  import torch
62
 
63
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
37
 
38
  #### Generate Embeddings
39
  ```python
40
+ from helical.models.helix_mrna import HelixmRNA, HelixmRNAConfig
41
  import torch
42
 
43
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
57
  #### Fine-Tuning
58
  Classification fine-tuning example:
59
  ```python
60
+ from helical.models.helix_mrna import HelixmRNA, HelixmRNAConfig
61
  import torch
62
 
63
  device = "cuda" if torch.cuda.is_available() else "cpu"