sunwaee commited on
Commit
eefdc66
1 Parent(s): f224f43

updated page content

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -182,6 +182,8 @@ def process_img(model, image, labels, caption: bool = True):
182
  # Page config
183
  st.set_page_config(layout="centered")
184
  st.title("Face Mask Detection")
 
 
185
 
186
  # Models drive ids
187
  ids = {
@@ -197,11 +199,6 @@ labels = load_labels()
197
  model_path = st.selectbox('Choose a model', options=[k for k in ids], index=0)
198
  model = load_model(model_path=model_path) if model_path != '' else None
199
 
200
- # Content
201
- st.title('Face Mask Detection')
202
- st.write('ResNet[18~152] trained for Face Mask Detection. ')
203
- st.markdown(f"__Labels:__ with_mask, without_mask, mask_weared_incorrect")
204
-
205
  # Display example selection
206
  index = st.number_input('', min_value=0, max_value=852, value=495, help='Choose an image. ')
207
 
 
182
  # Page config
183
  st.set_page_config(layout="centered")
184
  st.title("Face Mask Detection")
185
+ st.write('ResNet[18~152] trained for Face Mask Detection. ')
186
+ st.markdown(f"__Labels:__ with_mask, without_mask, mask_weared_incorrect")
187
 
188
  # Models drive ids
189
  ids = {
 
199
  model_path = st.selectbox('Choose a model', options=[k for k in ids], index=0)
200
  model = load_model(model_path=model_path) if model_path != '' else None
201
 
 
 
 
 
 
202
  # Display example selection
203
  index = st.number_input('', min_value=0, max_value=852, value=495, help='Choose an image. ')
204