m7mdal7aj commited on
Commit
67de2bd
1 Parent(s): 2296d83

Update my_model/tabs/run_inference.py

Browse files
Files changed (1) hide show
  1. my_model/tabs/run_inference.py +4 -3
my_model/tabs/run_inference.py CHANGED
@@ -63,7 +63,7 @@ class InferenceRunner(StateManager):
63
  image = Image.open(sample_image_path)
64
  image_for_display = self.resize_image(sample_image_path, 80, 80)
65
  st.image(image_for_display)
66
- if st.button(f'Select Sample Image {idx + 1}', key=f'sample_{idx}'):
67
  self.process_new_image(sample_image_path, image)
68
 
69
  def handle_image_upload(self) -> None:
@@ -122,7 +122,8 @@ class InferenceRunner(StateManager):
122
 
123
  if self.settings_changed or self.confidance_change:
124
  nested_col22.warning("Confidence level changed, please click 'Analyze Image'.")
125
- nested_col22.write(self.settings_changed, self.confidance_change)
 
126
 
127
  def display_question_answering_interface(self, image_key: str, image_data: Dict, nested_col22: st.columns) -> None:
128
  """
@@ -262,7 +263,7 @@ class InferenceRunner(StateManager):
262
 
263
 
264
  elif st.session_state.method == "Learning Visual Embeddings":
265
- self.col1.warning(f'Model using {st.session_state.method} is desgined but requires large scale data and multiple high-end GPUS, implementation will be explored in the future.')
266
 
267
 
268
  if self.is_model_loaded:
 
63
  image = Image.open(sample_image_path)
64
  image_for_display = self.resize_image(sample_image_path, 80, 80)
65
  st.image(image_for_display)
66
+ if st.button(f'Select Sample Image {idx + 1}', key=f'sample_{idx+1}'):
67
  self.process_new_image(sample_image_path, image)
68
 
69
  def handle_image_upload(self) -> None:
 
122
 
123
  if self.settings_changed or self.confidance_change:
124
  nested_col22.warning("Confidence level changed, please click 'Analyze Image'.")
125
+ nested_col22.write(self.settings_changed)
126
+ nested_col22.write(self.confidance_change)
127
 
128
  def display_question_answering_interface(self, image_key: str, image_data: Dict, nested_col22: st.columns) -> None:
129
  """
 
263
 
264
 
265
  elif st.session_state.method == "Learning Visual Embeddings":
266
+ self.col1.warning(f'Model using {st.session_state.method} is desgined but requires large scale data and multiple high-end GPUs, implementation will be explored in the future.')
267
 
268
 
269
  if self.is_model_loaded: