Update app.py

app.py

@@ -80,7 +80,11 @@ def inference(
 
     det_model_path = str(Path("models") / "text_det" / text_det)
     rec_model_path = str(Path("models") / "text_rec" / text_rec)
-    if "v2" in rec_model_path:
+    if (
+        "v2" in rec_model_path
+        or "korean" in rec_model_path
+        or "japan" in rec_model_path
+    ):
         rec_image_shape = [3, 32, 320]
     else:
         rec_image_shape = [3, 48, 320]
@@ -104,7 +108,7 @@ def inference(
         img, box_thresh=box_thresh, unclip_ratio=unclip_ratio, text_score=text_score
     )
     if not ocr_result or not infer_elapse:
-        return None, None, None, None
+        return None, None, None
 
     det_cost, cls_cost, rec_cost = infer_elapse
     elapse = f"- `det cost`: {det_cost:.5f}\n - `cls cost`: {cls_cost:.5f}\n - `rec cost`: {rec_cost:.5f}"
@@ -178,17 +182,18 @@ if __name__ == "__main__":
     ]
     select_rec = menu_rec.selectbox("Rec model:", rec_models)
 
-    img_file_buffer = st.file_uploader(
-        "Upload an image",
-        accept_multiple_files=False,
-        label_visibility="visible",
-        type=["png", "jpg", "jpeg", "bmp"],
-    )
-
-    if img_file_buffer:
-        image = Image.open(img_file_buffer)
-        img = np.array(image)
-        st.session_state["img"] = img
+    with st.form("my-form", clear_on_submit=True):
+        img_file_buffer = st.file_uploader(
+            "Upload an image",
+            accept_multiple_files=False,
+            label_visibility="visible",
+            type=["png", "jpg", "jpeg", "bmp"],
+        )
+        submit = st.form_submit_button("Upload")
+        if submit and img_file_buffer is not None:
+            image = Image.open(img_file_buffer)
+            img = np.array(image)
+            st.session_state["img"] = img
 
     if st.session_state["img"] is not None:
         out_img, out_json, elapse = inference(select_det, select_rec)