Spaces:
Sleeping
Sleeping
Update my_model/tabs/run_inference.py
Browse files
my_model/tabs/run_inference.py
CHANGED
@@ -112,7 +112,7 @@ def run_inference():
|
|
112 |
|
113 |
|
114 |
# need_model_reload = state_manager.check_settings_changed(method, detection_model, confidence_level) and state_manager.is_model_loaded()
|
115 |
-
|
116 |
button_label = "Reload Model" if st.session_state['kbvqa'] else "Load Model"
|
117 |
|
118 |
if method == "Fine-Tuned Model":
|
|
|
112 |
|
113 |
|
114 |
# need_model_reload = state_manager.check_settings_changed(method, detection_model, confidence_level) and state_manager.is_model_loaded()
|
115 |
+
|
116 |
button_label = "Reload Model" if st.session_state['kbvqa'] else "Load Model"
|
117 |
|
118 |
if method == "Fine-Tuned Model":
|