Add slider

main.py

@@ -16,19 +16,25 @@ def get_rec(points):
     return (min(xs), min(ys)), (max(xs), max(ys))
         
     
-def predict(image_input):
+def predict(image_input, score_threshold):
     draw_img = image_input.copy()
     print(image_input.shape)
     output = ocr.readtext(image_input)
-    for polygon in output['det_polygons']:
+    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)
+        draw_img = cv2.putText(draw_img, '%.3f' % score, p0)
     return draw_img
 
 def run():
     demo = gr.Interface(
         fn=predict,
-        inputs=gr.components.Image(),
+        inputs=[gr.components.Image(), gr.Slider(0, 1)],
         outputs=gr.components.Image(),
     )