Update README.md
Browse files
README.md
CHANGED
|
@@ -23,7 +23,6 @@ After installing CodonTransformer, you can use:
|
|
| 23 |
import torch
|
| 24 |
from transformers import AutoTokenizer, BigBirdForMaskedLM
|
| 25 |
from CodonTransformer.CodonPrediction import predict_dna_sequence
|
| 26 |
-
from CodonTransformer.CodonUtils import ORGANISM2ID
|
| 27 |
from CodonTransformer.CodonJupyter import format_model_output
|
| 28 |
DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 29 |
|
|
|
|
| 23 |
import torch
|
| 24 |
from transformers import AutoTokenizer, BigBirdForMaskedLM
|
| 25 |
from CodonTransformer.CodonPrediction import predict_dna_sequence
|
|
|
|
| 26 |
from CodonTransformer.CodonJupyter import format_model_output
|
| 27 |
DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 28 |
|