Update image2text_example.py

hezarai · Sep 20, 2023 · 8f2f486 · 8f2f486
1 parent 6a8ef5f
commit 8f2f486
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/examples/image2text_example.py b/examples/image2text_example.py
@@ -1,5 +1,6 @@
 from hezar import Model
 
-model = Model.load("hezarai/trocr-fa-v1")
-text = model.predict(["assets/ocr_example.jpg"])
+# model = Model.load("hezarai/trocr-fa-v1")  # TrOCR
+model = Model.load("hezarai/crnn-base-fa-64x256")  # CRNN
+text = model.predict("assets/ocr_example.jpg")
 print(text)