m7mdal7aj commited on
Commit
6d914c0
1 Parent(s): 9023a92

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -109,7 +109,7 @@ def run_inference():
109
 
110
  # Slider for confidence level
111
  confidence_level = st.slider(
112
- "Select Detection Confidence Level",
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 KBVQA 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.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