m7mdal7aj commited on
Commit
07be141
1 Parent(s): 33dfb0b

Update my_model/tabs/run_inference.py

Browse files
Files changed (1) hide show
  1. my_model/tabs/run_inference.py +6 -1
my_model/tabs/run_inference.py CHANGED
@@ -122,6 +122,7 @@ 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
 
126
  def display_question_answering_interface(self, image_key: str, image_data: Dict, nested_col22: st.columns) -> None:
127
  """
@@ -258,7 +259,11 @@ class InferenceRunner(StateManager):
258
 
259
  elif st.session_state.method == "In-Context Learning (n-shots)":
260
  self.col1.warning(f'Model using {st.session_state.method} is not deployed yet, will be ready later.')
261
- st.session_state['loading_in_progress'] = False
 
 
 
 
262
 
263
  if self.is_model_loaded:
264
  free_gpu_resources()
 
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
  """
 
259
 
260
  elif st.session_state.method == "In-Context Learning (n-shots)":
261
  self.col1.warning(f'Model using {st.session_state.method} is not deployed yet, will be ready later.')
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:
269
  free_gpu_resources()