Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -218,7 +218,7 @@ class InferenceRunner(StateManager):
         st.session_state.button_label = "Reload Model" if self.is_model_loaded and st.session_state.kbvqa.detection_model != st.session_state['detection_model'] else "Load Model"
       
         with self.col1:
-            if st.session_state.method == "Fine-Tuned Model":
+            if st.session_state.method == "7b-Fine-Tuned Model":
                 with st.container():
                     nested_col11, nested_col12 = st.columns([0.5, 0.5])
                     if nested_col11.button(st.session_state.button_label, on_click=self.disable_widgets, disabled=self.is_widget_disabled):
@@ -259,7 +259,7 @@ class InferenceRunner(StateManager):
                     st.session_state['loading_in_progress'] = False
                     st.session_state['model_loaded'] = True
                     
-            elif st.session_state.method == "In-Context Learning (n-shots)":
+            elif st.session_state.method == "13b-Fine-Tuned Model":
                 self.col1.warning(f'Model using {st.session_state.method} is not deployed yet, will be ready later.')