Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -48,7 +48,7 @@ class InferenceRunner(StateManager):
         """
         free_gpu_resources()
         answer = model.generate_answer(question, caption, detected_objects_str)
-        prompt_length  = modelcurrent_prompt_length
+        prompt_length  = model.current_prompt_length
         free_gpu_resources()
         return answer, prompt_length
 
@@ -113,17 +113,17 @@ class InferenceRunner(StateManager):
                         # Use the selected sample question or the custom question
                         question = custom_question if selected_question == "Custom question..." else selected_question
                         
-                        if not question:
-                            nested_col22.warning("Please select or enter a question.")
+                     #   if not question:
+                     #       nested_col22.warning("Please select or enter a question.")
+                     #   else:
+                        if question in [q for q, _ in qa_history]:
+                            nested_col22.warning("This question has already been answered.")
                         else:
-                            if question in [q for q, _ in qa_history]:
-                                nested_col22.warning("This question has already been answered.")
-                            else:
-                                if nested_col22.button('Get Answer', key=f'answer_{image_key}', disabled=self.is_widget_disabled):
+                            if nested_col22.button('Get Answer', key=f'answer_{image_key}', disabled=self.is_widget_disabled):
 
-                                    answer, prompt_length = self.answer_question(image_data['caption'], image_data['detected_objects_str'], question, kbvqa)
-                                    st.session_state['loading_in_progress'] = False
-                                    self.add_to_qa_history(image_key, question, answer, prompt_length)
+                                answer, prompt_length = self.answer_question(image_data['caption'], image_data['detected_objects_str'], question, kbvqa)
+                                st.session_state['loading_in_progress'] = False
+                                self.add_to_qa_history(image_key, question, answer, prompt_length)
         
                     # Display Q&A history and prompts lengths for each image
                     for num, (q, a, p) in enumerate(qa_history):