init

.gitignore

@@ -0,0 +1 @@
+ocr

app.py

@@ -0,0 +1,29 @@
+from doctr.io import DocumentFile
+from doctr.models import ocr_predictor
+from PIL import Image
+import gradio as gr
+
+# Load the model
+predictor = ocr_predictor(pretrained=True)
+
+def read_img(image):
+    # save image
+    image = image.fromarray(image)
+    image.save('image.jpg')
+    # Read the image
+    doc = DocumentFile.from_images(image)
+    # Run the model
+    result = predictor(doc)
+    # Return the result
+    return result
+
+app = gr.Interface(
+    title="Image OCR",
+    description="Extract text from images",
+    inputs=gr.inputs.Image(type="pil"),
+    outputs=gr.outputs.Textbox(),
+    fn=read_img
+)
+    
+
+app.launch(debug=True)