Add CRNN example to docs

hezarai · Sep 20, 2023 · ec89094 · ec89094
1 parent 8f2f486
commit ec89094
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 8 deletions.
diff --git a/README.md b/README.md
@@ -81,13 +81,19 @@ print(transcripts)
 - Image to Text (OCR)
 ```python
 from hezar import Model
-
+# OCR with TrOCR 
 model = Model.load("hezarai/trocr-fa-v1")
-text = model.predict(["examples/assets/ocr_example.jpg"])
-print(text)
+texts = model.predict(["examples/assets/ocr_example.jpg"])
+print(f"TrOCR Output: {texts}")
+
+# OCR with CRNN
+model = Model.load("hezarai/crnn-base-fa-64x256")
+texts = model.predict("examples/assets/ocr_example.jpg")
+print(f"CRNN Output: {texts}")
 ```
 ```
-{'texts': [' چه میشه کرد، باید صبر کنیم']}
+TrOCR Output: {'texts': [' چه میشه کرد، باید صبر کنیم']}
+CRNN Output: {'texts': ['چه میشه کرد، باید صبر کنیم']}
 ```
 We constantly keep working on adding and training new models and this section will hopefully be expanding every day ;)
 ### Word Embeddings

diff --git a/docs/get_started/quick_tour.md b/docs/get_started/quick_tour.md
@@ -46,13 +46,19 @@ print(transcripts)
 - **Image to Text (OCR)**
 ```python
 from hezar import Model
-
+# OCR with TrOCR 
 model = Model.load("hezarai/trocr-fa-v1")
-text = model.predict(["examples/assets/ocr_example.jpg"])
-print(text)
+texts = model.predict(["examples/assets/ocr_example.jpg"])
+print(f"TrOCR Output: {texts}")
+
+# OCR with CRNN
+model = Model.load("hezarai/crnn-base-fa-64x256")
+texts = model.predict("examples/assets/ocr_example.jpg")
+print(f"CRNN Output: {texts}")
 ```
 ```
-{'texts': [' چه میشه کرد، باید صبر کنیم']}
+TrOCR Output: {'texts': [' چه میشه کرد، باید صبر کنیم']}
+CRNN Output: {'texts': ['چه میشه کرد، باید صبر کنیم']}
 ```
 We constantly keep working on adding and training new models and this section will hopefully be expanding every day ;)
 ### Word Embeddings