Fixed problem that cannot upload custom image

app.py

@@ -58,16 +58,6 @@ def init_sidebar():
     )
     st.session_state["params"]["text_score"] = text_score
 
-    img_file_buffer = st.sidebar.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)
-
     with st.sidebar.container():
         img_path = image_select(
             label="Examples（click to select）:",
@@ -188,16 +178,29 @@ if __name__ == "__main__":
     ]
     select_rec = menu_rec.selectbox("Rec model:", rec_models)
 
-    out_img, out_json, elapse, only_txts = inference(select_det, select_rec)
-    if all(v is not None for v in [out_img, out_json, elapse]):
-        st.markdown("#### Visualize:")
-        st.image(out_img)
-
-        st.markdown("### Rec Result:")
-        st.markdown(elapse)
-        st.dataframe(out_json, use_container_width=True)
+    img_file_buffer = st.file_uploader(
+        "Upload an image",
+        accept_multiple_files=False,
+        label_visibility="visible",
+        type=["png", "jpg", "jpeg", "bmp"],
+    )
 
-        st.markdown("### Only Txts")
-        st.code(only_txts)
-    else:
-        tips("识别结果为空", wait_time=5, icon="⚠️")
+    if img_file_buffer:
+        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, only_txts = inference(select_det, select_rec)
+        if all(v is not None for v in [out_img, out_json, elapse]):
+            st.markdown("#### Visualize:")
+            st.image(out_img)
+
+            st.markdown("### Rec Result:")
+            st.markdown(elapse)
+            st.dataframe(out_json, use_container_width=True)
+
+            st.markdown("### Only Txts")
+            st.code(only_txts)
+        else:
+            tips("识别结果为空", wait_time=5, icon="⚠️")