Upload README.md with huggingface_hub

README.md

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