Update app.py

app.py

@@ -11,6 +11,8 @@ with st.sidebar:
     runtime_agent_device=st.radio("Choose runtime agent device combo",("N2X-ORCA1","TFLite-EdgeTPU","OpenVINO-CPU"),index=0,horizontal=True)
     activation_option=st.radio( 'Select activation function', ['relu6', 'silu'],horizontal=True)
     dataset_option=st.radio( 'Select a dataset option', ['coco', 'face','lp','car','hand'],horizontal=True)
+    show_labels=st.toggle('Show labels in output',value=True)
+    show_probabilities=st.toggle('Show probabilities in output',value=False)
 runtime_agent,device=runtime_agent_device.split('-')[0],runtime_agent_device.split('-')[1]
 model_options=zoo.list_models(device=device,runtime=runtime_agent)
 st.header('Choose and Run a Model')
@@ -25,10 +27,13 @@ with st.form("model_form"):
     uploaded_file=st.file_uploader('input image')
     submitted = st.form_submit_button("Submit")
     if submitted:
-        model=zoo.load_model(model_name)
-        model.overlay_font_scale=3
-        model.overlay_line_width=6
-        model.image_backend='pil'        
+        model=zoo.load_model(model_name,
+                             overlay_show_labels=show_labels,
+                             overlay_show_probabilities=show_probabilities,
+                             overlay_font_scale=3,
+                             overlay_line_width=6
+                             image_backend='pil'                                     
+                            )
         if model.output_postprocess_type=='PoseDetection':
             model.overlay_show_labels=False
         st.write("Model loaded successfully")