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