Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -17,14 +17,9 @@ 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",
-            "Files/sample7.jpg"
-        ]
+            "Files/sample1.jpg", "Files/sample2.jpg", "Files/sample3.jpg"]
 
     def answer_question(self, caption, detected_objects_str, question, model):
         free_gpu_resources()
@@ -74,7 +69,6 @@ class InferenceRunner(StateManager):
 
 
     def run_inference(self):
-        
         self.set_up_widgets()
         st.session_state['settings_changed'] = self.has_state_changed()
         if st.session_state['settings_changed']: