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="⚠️")