Update app.py

app.py

@@ -117,12 +117,7 @@ def run_inference():
         index=0  # Default to the first option
     )
 
-
-
-
-    
     # Initialize session state for the model
-
     if method == "Fine-Tuned Model":
         if 'kbvqa' not in st.session_state:
             st.session_state['kbvqa'] = None
@@ -137,28 +132,23 @@ def run_inference():
     
             if st.session_state['kbvqa']:
                 st.write("Model is ready for 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']:
             image_qa_app(st.session_state['kbvqa'])
 
     else: 
         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
-    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
 def main():