Update app.py

app.py

@@ -20,8 +20,8 @@ def rectangle(image, result):
 # main title
 st.title("Recognize text and locations from flowchart")
 
-# subtitle
-st.markdown("## FlowchartOCR")
+# # subtitle
+# st.markdown("## FlowchartOCR")
 
 # upload image file
 file = st.file_uploader(label = "Upload Image", type=['png', 'jpg', 'jpeg'])
@@ -44,6 +44,7 @@ if file is not None:
         pred_confidence = result[idx][2]
         if(pred_confidence>0.6):
             textdic_easyocr[pred_text] = {}
+            textdic_easyocr[pred_text]['location'] = pred_coor
             textdic_easyocr[pred_text]['pred_confidence'] = pred_confidence
 
     # create a data frame which shows the predicted text and prediction confidence