Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -98,9 +98,11 @@ def run_inference():
     st.title("Run Inference")
     state_manager.initialize_state()
     state_manager.set_up_widgets()
+    st.session_state.button_label = "Reload Model" if state_manager.is_model_loaded() and state_manager.has_state_changed() else "Load Model"
    # state_manager.display_session_state()
     state_manager.display_model_settings()
-    button_label = "Reload Model" if state_manager.is_model_loaded() and state_manager.has_state_changed() else "Load Model"
+    
+    
     if st.session_state.method == "Fine-Tuned Model":
         if st.button(button_label):
             if button_label == "Load Model":