Update app.py
Browse files
app.py
CHANGED
@@ -109,7 +109,7 @@ def run_inference():
|
|
109 |
|
110 |
# Slider for confidence level
|
111 |
confidence_level = st.slider(
|
112 |
-
"Select
|
113 |
min_value=0.1,
|
114 |
max_value=0.9,
|
115 |
value=default_confidence,
|
@@ -125,11 +125,12 @@ def run_inference():
|
|
125 |
st.session_state['kbvqa'] = None
|
126 |
|
127 |
# Button to load KBVQA models
|
128 |
-
if st.button('Load
|
129 |
if st.session_state['kbvqa'] is not None:
|
130 |
st.write("Model already loaded.")
|
131 |
else:
|
132 |
# Call the function to load models and show progress
|
|
|
133 |
st.session_state['kbvqa'] = prepare_kbvqa_model(detection_model)
|
134 |
st.session_state['kbvqa'].detection_confidence = confidence_level
|
135 |
|
|
|
109 |
|
110 |
# Slider for confidence level
|
111 |
confidence_level = st.slider(
|
112 |
+
"Select minimum detection confidence level",
|
113 |
min_value=0.1,
|
114 |
max_value=0.9,
|
115 |
value=default_confidence,
|
|
|
125 |
st.session_state['kbvqa'] = None
|
126 |
|
127 |
# Button to load KBVQA models
|
128 |
+
if st.button('Load Model'):
|
129 |
if st.session_state['kbvqa'] is not None:
|
130 |
st.write("Model already loaded.")
|
131 |
else:
|
132 |
# Call the function to load models and show progress
|
133 |
+
st.text("Loading the model will take no more than a few minutes . .")
|
134 |
st.session_state['kbvqa'] = prepare_kbvqa_model(detection_model)
|
135 |
st.session_state['kbvqa'].detection_confidence = confidence_level
|
136 |
|