Update app.py

app.py

@@ -1,7 +1,14 @@
 import gradio as gr
+from craft_hw_ocr import OCR
 
-def greet(name):
-    return "Hello " + name + "!!"
+trocr = OCR.load_TrOCRmodel()
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+def do_ocr(inp):
+  img, results = OCR.craft_detection(inp)
+  bboxes, text = OCR.text_recoginition(img, results, trocr[0], trocr[1])
+  return OCR.visualize(img, results), text
+  
+inputs = gr.inputs.Image()
+o1 = gr.outputs.Image()
+o2 = gr.outputs.Textbox()
+gr.Interface(fn=predict, inputs=inputs, outputs=[o1, o2]).launch()