m7mdal7aj commited on
Commit
a362b17
·
verified ·
1 Parent(s): ab9de87

Update my_model/tabs/run_inference.py

Browse files
Files changed (1) hide show
  1. my_model/tabs/run_inference.py +2 -1
my_model/tabs/run_inference.py CHANGED
@@ -89,7 +89,8 @@ class InferenceRunner(StateManager):
89
  nested_col22.text("Please click 'Analyze Image'..")
90
  free_gpu_resources()
91
  with nested_col22:
92
- if st.button('Analyze Image', key=f'analyze_{image_key}', on_click=self.disable_widgets, disabled=self.is_widget_disabled):
 
93
  st.text("AAAAAAAAAAAAAAAAAAAAA")
94
  caption, detected_objects_str, image_with_boxes = self.analyze_image(image_data['image'])
95
  st.text("BBBBBBBBBBBBBBBBBBBB")
 
89
  nested_col22.text("Please click 'Analyze Image'..")
90
  free_gpu_resources()
91
  with nested_col22:
92
+ analyze_button_key = f'analyze_{image_key}_{st.session_state.detection_model}_{st.session_state.confidence_level}' # unique key for each click
93
+ if st.button('Analyze Image', key=analyze_button_key, on_click=self.disable_widgets, disabled=self.is_widget_disabled):
94
  st.text("AAAAAAAAAAAAAAAAAAAAA")
95
  caption, detected_objects_str, image_with_boxes = self.analyze_image(image_data['image'])
96
  st.text("BBBBBBBBBBBBBBBBBBBB")