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