Update app.py

app.py

@@ -147,17 +147,17 @@ def run_inference():
 
     # 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
+    if st.session_state['kbvqa']:
+        # 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