first commit
Browse files- .gradio/certificate.pem +31 -0
- app.py +62 -0
- images/demo1.png +0 -0
- requirements.txt +4 -0
.gradio/certificate.pem
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
-----BEGIN CERTIFICATE-----
|
| 2 |
+
MIIFazCCA1OgAwIBAgIRAIIQz7DSQONZRGPgu2OCiwAwDQYJKoZIhvcNAQELBQAw
|
| 3 |
+
TzELMAkGA1UEBhMCVVMxKTAnBgNVBAoTIEludGVybmV0IFNlY3VyaXR5IFJlc2Vh
|
| 4 |
+
cmNoIEdyb3VwMRUwEwYDVQQDEwxJU1JHIFJvb3QgWDEwHhcNMTUwNjA0MTEwNDM4
|
| 5 |
+
WhcNMzUwNjA0MTEwNDM4WjBPMQswCQYDVQQGEwJVUzEpMCcGA1UEChMgSW50ZXJu
|
| 6 |
+
ZXQgU2VjdXJpdHkgUmVzZWFyY2ggR3JvdXAxFTATBgNVBAMTDElTUkcgUm9vdCBY
|
| 7 |
+
MTCCAiIwDQYJKoZIhvcNAQEBBQADggIPADCCAgoCggIBAK3oJHP0FDfzm54rVygc
|
| 8 |
+
h77ct984kIxuPOZXoHj3dcKi/vVqbvYATyjb3miGbESTtrFj/RQSa78f0uoxmyF+
|
| 9 |
+
0TM8ukj13Xnfs7j/EvEhmkvBioZxaUpmZmyPfjxwv60pIgbz5MDmgK7iS4+3mX6U
|
| 10 |
+
A5/TR5d8mUgjU+g4rk8Kb4Mu0UlXjIB0ttov0DiNewNwIRt18jA8+o+u3dpjq+sW
|
| 11 |
+
T8KOEUt+zwvo/7V3LvSye0rgTBIlDHCNAymg4VMk7BPZ7hm/ELNKjD+Jo2FR3qyH
|
| 12 |
+
B5T0Y3HsLuJvW5iB4YlcNHlsdu87kGJ55tukmi8mxdAQ4Q7e2RCOFvu396j3x+UC
|
| 13 |
+
B5iPNgiV5+I3lg02dZ77DnKxHZu8A/lJBdiB3QW0KtZB6awBdpUKD9jf1b0SHzUv
|
| 14 |
+
KBds0pjBqAlkd25HN7rOrFleaJ1/ctaJxQZBKT5ZPt0m9STJEadao0xAH0ahmbWn
|
| 15 |
+
OlFuhjuefXKnEgV4We0+UXgVCwOPjdAvBbI+e0ocS3MFEvzG6uBQE3xDk3SzynTn
|
| 16 |
+
jh8BCNAw1FtxNrQHusEwMFxIt4I7mKZ9YIqioymCzLq9gwQbooMDQaHWBfEbwrbw
|
| 17 |
+
qHyGO0aoSCqI3Haadr8faqU9GY/rOPNk3sgrDQoo//fb4hVC1CLQJ13hef4Y53CI
|
| 18 |
+
rU7m2Ys6xt0nUW7/vGT1M0NPAgMBAAGjQjBAMA4GA1UdDwEB/wQEAwIBBjAPBgNV
|
| 19 |
+
HRMBAf8EBTADAQH/MB0GA1UdDgQWBBR5tFnme7bl5AFzgAiIyBpY9umbbjANBgkq
|
| 20 |
+
hkiG9w0BAQsFAAOCAgEAVR9YqbyyqFDQDLHYGmkgJykIrGF1XIpu+ILlaS/V9lZL
|
| 21 |
+
ubhzEFnTIZd+50xx+7LSYK05qAvqFyFWhfFQDlnrzuBZ6brJFe+GnY+EgPbk6ZGQ
|
| 22 |
+
3BebYhtF8GaV0nxvwuo77x/Py9auJ/GpsMiu/X1+mvoiBOv/2X/qkSsisRcOj/KK
|
| 23 |
+
NFtY2PwByVS5uCbMiogziUwthDyC3+6WVwW6LLv3xLfHTjuCvjHIInNzktHCgKQ5
|
| 24 |
+
ORAzI4JMPJ+GslWYHb4phowim57iaztXOoJwTdwJx4nLCgdNbOhdjsnvzqvHu7Ur
|
| 25 |
+
TkXWStAmzOVyyghqpZXjFaH3pO3JLF+l+/+sKAIuvtd7u+Nxe5AW0wdeRlN8NwdC
|
| 26 |
+
jNPElpzVmbUq4JUagEiuTDkHzsxHpFKVK7q4+63SM1N95R1NbdWhscdCb+ZAJzVc
|
| 27 |
+
oyi3B43njTOQ5yOf+1CceWxG1bQVs5ZufpsMljq4Ui0/1lvh+wjChP4kqKOJ2qxq
|
| 28 |
+
4RgqsahDYVvTH9w7jXbyLeiNdd8XM2w9U/t7y0Ff/9yi0GE44Za4rF2LN9d11TPA
|
| 29 |
+
mRGunUHBcnWEvgJBQl9nJEiU0Zsnvgc/ubhPgXRR4Xq37Z0j4r7g1SgEEzwxA57d
|
| 30 |
+
emyPxgcYxn/eR44/KJ4EBs+lVDR3veyJm+kXQ99b21/+jh5Xos1AnX5iItreGCc=
|
| 31 |
+
-----END CERTIFICATE-----
|
app.py
ADDED
|
@@ -0,0 +1,62 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import gradio as gr
|
| 2 |
+
from kiri_ocr import OCR
|
| 3 |
+
import os
|
| 4 |
+
from PIL import Image, ImageDraw
|
| 5 |
+
import numpy as np
|
| 6 |
+
|
| 7 |
+
# Initialize OCR
|
| 8 |
+
try:
|
| 9 |
+
print("Loading Kiri OCR model...")
|
| 10 |
+
ocr = OCR()
|
| 11 |
+
print("Model loaded successfully")
|
| 12 |
+
except Exception as e:
|
| 13 |
+
print(f"Error loading model: {e}")
|
| 14 |
+
ocr = None
|
| 15 |
+
|
| 16 |
+
def process_image(image_path):
|
| 17 |
+
if ocr is None:
|
| 18 |
+
return None, "Error: OCR model failed to load."
|
| 19 |
+
|
| 20 |
+
if image_path is None:
|
| 21 |
+
return None, "Please upload an image."
|
| 22 |
+
|
| 23 |
+
try:
|
| 24 |
+
# extract_text returns (text, results)
|
| 25 |
+
text, results = ocr.extract_text(image_path)
|
| 26 |
+
|
| 27 |
+
# Open image for drawing
|
| 28 |
+
img = Image.open(image_path)
|
| 29 |
+
# Handle transparency if needed, convert to RGB
|
| 30 |
+
if img.mode != 'RGB':
|
| 31 |
+
img = img.convert('RGB')
|
| 32 |
+
|
| 33 |
+
draw = ImageDraw.Draw(img)
|
| 34 |
+
|
| 35 |
+
# Draw boxes
|
| 36 |
+
for item in results:
|
| 37 |
+
if 'box' in item:
|
| 38 |
+
x, y, w, h = item['box']
|
| 39 |
+
draw.rectangle([x, y, x + w, y + h], outline="red", width=3)
|
| 40 |
+
|
| 41 |
+
# Convert PIL image to numpy array for Gradio output
|
| 42 |
+
return np.array(img), text
|
| 43 |
+
|
| 44 |
+
except Exception as e:
|
| 45 |
+
return None, f"Error during extraction: {str(e)}"
|
| 46 |
+
|
| 47 |
+
# Define the Gradio interface
|
| 48 |
+
iface = gr.Interface(
|
| 49 |
+
fn=process_image,
|
| 50 |
+
inputs=gr.Image(type="filepath", label="Upload Image"),
|
| 51 |
+
outputs=[
|
| 52 |
+
gr.Image(label="Detected Text"), # Default type is numpy
|
| 53 |
+
gr.Textbox(label="Extracted Text")
|
| 54 |
+
],
|
| 55 |
+
title="Kiri OCR Demo",
|
| 56 |
+
description="Upload an image to extract English and Khmer text using Kiri OCR. The detected text regions are highlighted in the output image.\n\nPowered by [kiri-ocr](https://github.com/mrrtmob/kiri-ocr).",
|
| 57 |
+
theme="default"
|
| 58 |
+
)
|
| 59 |
+
|
| 60 |
+
# Launch the app
|
| 61 |
+
if __name__ == "__main__":
|
| 62 |
+
iface.launch()
|
images/demo1.png
ADDED
|
requirements.txt
ADDED
|
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
kiri-ocr
|
| 2 |
+
gradio
|
| 3 |
+
numpy
|
| 4 |
+
Pillow
|