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 |
|