Spaces:
Sleeping
Sleeping
Update my_model/tabs/run_inference.py
Browse files
my_model/tabs/run_inference.py
CHANGED
|
@@ -112,14 +112,12 @@ def run_inference():
|
|
| 112 |
slider_key_name='confidence_level'
|
| 113 |
)
|
| 114 |
|
| 115 |
-
|
| 116 |
-
state_manager.display_session_state()
|
| 117 |
-
|
| 118 |
# state_manager.update_model_settings(detection_model=detection_model, confidence_level=confidence_level, selected_method=method)
|
| 119 |
|
| 120 |
|
| 121 |
# need_model_reload = state_manager.check_settings_changed(method, detection_model, confidence_level) and state_manager.is_model_loaded()
|
| 122 |
-
|
| 123 |
|
| 124 |
button_label = "Reload Model" if st.session_state['kbvqa'] is not None else "Load Model"
|
| 125 |
|
|
@@ -135,8 +133,7 @@ def run_inference():
|
|
| 135 |
st.write("Model is ready for inference.")
|
| 136 |
|
| 137 |
if state_manager.is_model_loaded():
|
| 138 |
-
|
| 139 |
-
state_manager.display_session_state()
|
| 140 |
image_qa_app(state_manager.get_model())
|
| 141 |
|
| 142 |
else:
|
|
|
|
| 112 |
slider_key_name='confidence_level'
|
| 113 |
)
|
| 114 |
|
| 115 |
+
state_manager.display_model_settings()
|
|
|
|
|
|
|
| 116 |
# state_manager.update_model_settings(detection_model=detection_model, confidence_level=confidence_level, selected_method=method)
|
| 117 |
|
| 118 |
|
| 119 |
# need_model_reload = state_manager.check_settings_changed(method, detection_model, confidence_level) and state_manager.is_model_loaded()
|
| 120 |
+
|
| 121 |
|
| 122 |
button_label = "Reload Model" if st.session_state['kbvqa'] is not None else "Load Model"
|
| 123 |
|
|
|
|
| 133 |
st.write("Model is ready for inference.")
|
| 134 |
|
| 135 |
if state_manager.is_model_loaded():
|
| 136 |
+
|
|
|
|
| 137 |
image_qa_app(state_manager.get_model())
|
| 138 |
|
| 139 |
else:
|