Sanjayraju30 commited on
Commit
7b6f5c9
Β·
verified Β·
1 Parent(s): fec538c

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +3 -3
src/streamlit_app.py CHANGED
@@ -16,7 +16,7 @@ if "image_data" not in st.session_state:
16
  if st.button("πŸ” Clear / Retake Photo"):
17
  st.session_state.image_captured = False
18
  st.session_state.image_data = None
19
- st.experimental_rerun()
20
 
21
  # βœ… Show camera only if no image captured
22
  if not st.session_state.image_captured:
@@ -25,7 +25,7 @@ if not st.session_state.image_captured:
25
  if img_data:
26
  st.session_state.image_data = img_data
27
  st.session_state.image_captured = True
28
- st.experimental_rerun()
29
 
30
  # βœ… If image is already captured
31
  if st.session_state.image_captured and st.session_state.image_data:
@@ -50,7 +50,7 @@ if st.session_state.image_captured and st.session_state.image_data:
50
  if st.button("πŸ” Retake Photo"):
51
  st.session_state.image_captured = False
52
  st.session_state.image_data = None
53
- st.experimental_rerun()
54
  st.stop()
55
 
56
  st.success(f"βœ… Detected Weight: {weight} g (Confidence: {int(confidence)}%)")
 
16
  if st.button("πŸ” Clear / Retake Photo"):
17
  st.session_state.image_captured = False
18
  st.session_state.image_data = None
19
+ st.rerun()
20
 
21
  # βœ… Show camera only if no image captured
22
  if not st.session_state.image_captured:
 
25
  if img_data:
26
  st.session_state.image_data = img_data
27
  st.session_state.image_captured = True
28
+ st.rerun()
29
 
30
  # βœ… If image is already captured
31
  if st.session_state.image_captured and st.session_state.image_data:
 
50
  if st.button("πŸ” Retake Photo"):
51
  st.session_state.image_captured = False
52
  st.session_state.image_data = None
53
+ st.rerun()
54
  st.stop()
55
 
56
  st.success(f"βœ… Detected Weight: {weight} g (Confidence: {int(confidence)}%)")