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