Spaces:
Sleeping
Sleeping
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()
|