Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -104,7 +104,7 @@ def run_inference():
     
     
     if st.session_state.method == "Fine-Tuned Model":
-        if st.button(button_label):
+        if st.button(st.session_state.button_label):
             if button_label == "Load Model":
                 if state_manager.is_model_loaded():
                     st.text("Model already loaded and no settings were changed:)")