Rename NemoRetriverOCR to NemotronOCR

README.md

@@ -212,9 +212,9 @@ Output is saved next to your input image as `<name>-annotated.<ext>` on the host
 3. Run the model using the following code:
 
 ```python
-from nemotron_ocr.inference.pipeline import NemoRetrieverOCR
+from nemotron_ocr.inference.pipeline import NemotronOCR
 
-ocr = NemoRetrieverOCR()
+ocr = NemotronOCR()
 
 predictions = ocr("ocr-example-input-1.png")
 

example.py

@@ -4,11 +4,11 @@
 
 import argparse
 
-from nemotron_ocr.inference.pipeline import NemoRetrieverOCR
+from nemotron_ocr.inference.pipeline import NemotronOCR
 
 
 def main(image_path, merge_level, no_visualize, model_dir):
-    ocr_pipeline = NemoRetrieverOCR()
+    ocr_pipeline = NemotronOCR()
 
     predictions = ocr_pipeline(image_path, merge_level=merge_level, visualize=not no_visualize)
 

nemotron-ocr/src/nemotron_ocr/inference/pipeline.py

@@ -37,7 +37,7 @@ MERGE_LEVELS = {"word", "sentence", "paragraph"}
 DEFAULT_MERGE_LEVEL = "paragraph"
 
 
-class NemoRetrieverOCR:
+class NemotronOCR:
     """
     A high-level pipeline for performing OCR on images.
     """

quickstart.md

@@ -29,12 +29,12 @@ uv run python -c "import nemotron_ocr; import nemotron_ocr_cpp"
 
 ## Usage
 
-`nemotron_ocr.inference.pipeline.NemoRetrieverOCR` is the main entry point for performing OCR inference; it can be used to iterate over predictions for a given input image:
+`nemotron_ocr.inference.pipeline.NemotronOCR` is the main entry point for performing OCR inference; it can be used to iterate over predictions for a given input image:
 
 ```python
-from nemotron_ocr.inference.pipeline import NemoRetrieverOCR
+from nemotron_ocr.inference.pipeline import NemotronOCR
 
-ocr = NemoRetrieverOCR()
+ocr = NemotronOCR()
 
 predictions = ocr("ocr-example-input-1.png")