Update main.py

main.py

@@ -22,30 +22,14 @@ def get_rec(points):
     return (min(xs), min(ys)), (max(xs), max(ys))
         
     
-def predict(image_input, score_threshold):
-    draw_img = image_input.copy()
-    print('image shape', image_input.shape, file=sys.stderr)
-
-    try:
-        return ocr.readtext(image_input)
-    except:
-        output = dict(det_polygons=[], det_scores=[])
-
-    polygons = output['det_polygons']
-    scores = output['det_scores']
-    for polygon, score in zip(polygons, scores):
-        if score < score_threshold:
-            continue
-        p0, p1 = get_rec([int(i) for i in polygon])
-        draw_img = cv2.rectangle(draw_img, p0, p1, (255,0,0), 2)
-
-    return draw_img, output
+def predict(image_input):
+    return ocr.readtext(image_input)
 
 def run():
     demo = gr.Interface(
         fn=predict,
-        inputs=[gr.components.Image(), gr.Slider(0, 1, 0.8)],
-        outputs=[gr.components.Image(), gr.JSON()],
+        inputs=gr.components.Image(),
+        outputs=gr.JSON(),
     )
 
     demo.launch(server_name="0.0.0.0", server_port=7860)