Fix get model cache

app.py

@@ -24,7 +24,7 @@ st.subheader(' Identify valves and pumps with deep learning model ', divider='ra
 st.caption('Developed by Deep Drawings Co.')
 
 @st.cache_resource(show_spinner=False)
-def get_model():
+def get_model(postprocess_match_threshold):
     yolov8_model_path = download_from_hub('DanielCerda/pid_yolov8')
     detection_model = AutoDetectionModel.from_pretrained(
         model_type='yolov8',
@@ -131,7 +131,7 @@ with col2:
 if submit:
     # perform prediction
     with st.spinner(text="Downloading model weights ... "):
-        detection_model = get_model()
+        detection_model = get_model(postprocess_match_threshold)
         
     image_size = 4960
 
@@ -151,7 +151,7 @@ if submit:
 
 st.write('##')
 
-col1, col2, col3 = st.columns([3, 1, 1], gap='medium')
+col1, col2, col3 = st.columns([3, 1, 1], gap='small')
 with col1:
     st.markdown(f"#### Object Detection Result")
     with st.container(border = True):