Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -15,7 +15,11 @@ from my_model.state_manager import StateManager
 
 class InferenceRunner(StateManager):
     def __init__(self):
+        
         super().__init__()
+        st.title("Run Inference")
+        st.write("Please note that this is not a general purpose mode, it is specifically trained on [OK-VQA Dataset](https://okvqa.allenai.org/) and desgined to give short and direct answers to the given questions about the given image")
+        self.initialize_state()
         self.sample_images = [
             "Files/sample1.jpg", "Files/sample2.jpg", "Files/sample3.jpg",
             "Files/sample4.jpg", "Files/sample5.jpg", "Files/sample6.jpg",
@@ -70,9 +74,7 @@ class InferenceRunner(StateManager):
 
 
     def run_inference(self):
-        st.title("Run Inference")
-        st.write("Please note that this is not a general purpose mode, it is specifically trained on [OK-VQA Dataset](https://okvqa.allenai.org/) and desgined to give short and direct answers to the given questions about the given image")
-        self.initialize_state()
+        
         self.set_up_widgets()
         st.session_state['settings_changed'] = self.has_state_changed()
         if st.session_state['settings_changed']: