Create app.py

app.py

@@ -0,0 +1,43 @@
+import os
+from doctr.io import DocumentFile
+from doctr.models import ocr_predictor
+import gradio as gr
+
+os.environ['USE_TORCH'] = '1'
+
+# ใช้โมเดล OCR ภาษาอังกฤษที่มีใน Doctr
+predictor = ocr_predictor(pretrained=True)
+
+title = "English OCR"
+description = """Upload an image to extract English text!"""
+
+def ocr(img):
+    img.save("out.jpg")
+    doc = DocumentFile.from_images("out.jpg")
+    output = predictor(doc)
+    res = ""
+    for page in output.pages:
+        for block in page.blocks:
+            for line in block.lines:
+                for word in line.words:
+                    res += word.value + " "
+            res += "\n"
+        res += "\n"
+    _output_name = "RESULT_OCR.txt"
+    with open(_output_name, "w", encoding="utf-8", errors="ignore") as f:
+        f.write(res)
+    return res, _output_name
+
+demo = gr.Interface(
+    fn=ocr,
+    inputs=[gr.Image(type="pil")],
+    outputs=[
+        gr.Textbox(lines=20, label="Full Text"),
+        gr.File(label="Download OCR Results")
+    ],
+    title=title,
+    description=description,
+    examples=[["Examples/english_sample.jpg"]]
+)
+
+demo.launch(debug=True)