Upload README.md with huggingface_hub

README.md

@@ -89,9 +89,7 @@ Opens a web UI at http://127.0.0.1:7860 where you can test and finetune on your
 ## Usage (Python)
 
 ```python
-from needle.model.run import load_checkpoint, generate
-from needle.model.architecture import EncoderDecoderTransformer
-from needle.dataset.dataset import get_tokenizer
+from needle import load_checkpoint, generate, EncoderDecoderTransformer, get_tokenizer
 
 params, config = load_checkpoint("checkpoints/needle.pkl")
 model = EncoderDecoderTransformer(config)