Add application file

README.md

@@ -1,13 +1,37 @@
 ---
-title: CRAFT
-emoji: 🐢
-colorFrom: yellow
-colorTo: green
+title: CRAFT OCR
+emoji: 👁
+colorFrom: pink
+colorTo: purple
 sdk: gradio
-sdk_version: 2.8.10
 app_file: app.py
 pinned: false
-license: mit
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces#reference
+# Configuration
+
+`title`: _string_  
+Display title for the Space
+
+`emoji`: _string_  
+Space emoji (emoji-only character allowed)
+
+`colorFrom`: _string_  
+Color for Thumbnail gradient (red, yellow, green, blue, indigo, purple, pink, gray)
+
+`colorTo`: _string_  
+Color for Thumbnail gradient (red, yellow, green, blue, indigo, purple, pink, gray)
+
+`sdk`: _string_  
+Can be either `gradio` or `streamlit`
+
+`sdk_version` : _string_  
+Only applicable for `streamlit` SDK.  
+See [doc](https://hf.co/docs/hub/spaces) for more info on supported versions.
+
+`app_file`: _string_  
+Path to your main application file (which contains either `gradio` or `streamlit` Python code).  
+Path is relative to the root of the repository.
+
+`pinned`: _boolean_  
+Whether the Space stays on top of your list.

app.py

@@ -0,0 +1,20 @@
+import gradio as gr
+from craft_hw_ocr import OCR
+
+ocr = OCR.load_models()
+
+def do_ocr(inp):
+  img, results = OCR.detection(inp, ocr[2])
+  bboxes, text = OCR.recoginition(img, results, ocr[0], ocr[1])
+  return OCR.visualize(img, results), text
+  
+inputs = gr.inputs.Image()
+o1 = gr.outputs.Image()
+o2 = gr.outputs.Textbox()
+
+title = "CRAFT-OCR"
+description = "OCR of both handwriting and printed text using CRAFT Text detector and TrOCR recognition, detection of lines and extraction of them are happening here because TrOCR pre-trained models are modelled on IAM lines dataset and the same needs to be implemented here."
+examples=[['example_1.png'],['example_2.jpg']]
+
+article = "<p style='text-align: center'><a href='https://github.com/Vishnunkumar/craft_hw_ocr' target='_blank'>craft_hw_ocr</a></p><p style='text-align: center'> <p style='text-align: center'><a href='https://github.com/fcakyon/craft-text-detector' target='_blank'>craft-text-detector</a></p><p style='text-align: center'>"
+gr.Interface(fn=do_ocr, inputs=inputs, outputs=[o1, o2], title=title, description=description, article=article, examples=examples, enable_queue=True).launch()

packages.txt

@@ -0,0 +1 @@
+python3-opencv

requirements.txt

@@ -0,0 +1,3 @@
+craft-hw-ocr==1.1
+gradio
+opencv-python-headless==4.5.5.62