Update app.py

app.py

@@ -92,7 +92,7 @@ def image_qa_app(kbvqa):
             st.session_state['answer_in_progress'] = True
             answer = answer_question(st.session_state['processed_image'], question, model=kbvqa)
             st.session_state['qa_history'].append((question, answer))
-            st.session_state['question'] = ''
+            
 
     # Display all Q&A
     for q, a in st.session_state['qa_history']:
@@ -117,17 +117,7 @@ def run_inference():
         index=0  # Default to the first option
     )
 
-    # Set default confidence based on the selected model
-    default_confidence = 0.2 if detection_model == "yolov5" else 0.4
 
-    # Slider for confidence level
-    confidence_level = st.slider(
-        "Select Detection Confidence Level",
-        min_value=0.1,
-        max_value=0.9,
-        value=default_confidence,
-        step=0.1
-    )
 
 
     
@@ -152,7 +142,22 @@ def run_inference():
             image_qa_app(st.session_state['kbvqa'])
 
     else: 
-        st.write('Model is not ready for inference yet')
+        st.write(f'{method} model is not ready for inference yet')
+        
+
+    # Set default confidence based on the selected model
+    default_confidence = 0.2 if detection_model == "yolov5" else 0.4
+
+    # Slider for confidence level
+    confidence_level = st.slider(
+        "Select Detection Confidence Level",
+        min_value=0.1,
+        max_value=0.9,
+        value=default_confidence,
+        step=0.1
+    )
+
+    st.session_state['kbvqa'].detection_confidence = confidence_level
 
             
 # Main function