Update my_model/tabs/run_inference.py
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
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()
|