Update app.py

app.py

@@ -1,36 +1,35 @@
 import os
-os.environ['USE_TORCH'] = '1'
-
 from doctr.io import DocumentFile
-from doctr.models import ocr_predictor
+from doctr.models import ocr_predictor, from_hub
 import gradio as gr
-from PIL import Image
 
-predictor = ocr_predictor(pretrained=True)
+os.environ['USE_TORCH'] = '1'
+reco_model = from_hub('ayymen/crnn_mobilenet_v3_large_gen_hw')
+predictor = ocr_predictor(reco_arch=reco_model, pretrained=True)
 
-title="DocTR OCR (PDL Demo)"
-description="Upload an image to get the OCR results !"
+title = "Tifinagh OCR"
+description = "Upload an image to get the OCR results !"
 
-def greet(img):
+def ocr(img):
     img.save("out.jpg")
     doc = DocumentFile.from_images("out.jpg")
-    output=predictor(doc)
-    res=""
+    output = predictor(doc)
+    res = ""
     for obj in output.pages:
-      for obj1 in obj.blocks:
-        for obj2 in obj1.lines:
-          for obj3 in obj2.words:
-            res=res + " " + obj3.value
-        res=res + "\n"
-      res=res + "\n"
+        for obj1 in obj.blocks:
+            for obj2 in obj1.lines:
+                for obj3 in obj2.words:
+                    res=res + " " + obj3.value
+            res=res + "\n"
+        res=res + "\n\n"
     _output_name = "RESULT_OCR.txt"
-    open(_output_name, 'w').close() # clear file
+    open(_output_name, 'w', encoding="utf-8").close() # clear file
     with open(_output_name, "w", encoding="utf-8", errors="ignore") as f:
         f.write(res)
         print("Writing into file")
     return res, _output_name
 
-demo = gr.Interface(fn=greet,
+demo = gr.Interface(fn=ocr,
                     inputs=gr.Image(type="pil"),
                     outputs=["text", "file"],
                     title=title,
@@ -38,4 +37,4 @@ demo = gr.Interface(fn=greet,
                     examples=[["Examples/Book.png"],["Examples/News.png"],["Examples/Manuscript.jpg"],["Examples/Files.jpg"]]
                     )
 
-demo.launch(debug=True)
+demo.launch(debug=True)