Update app.py

app.py

@@ -10,10 +10,15 @@ st.header('Specify Model Options Below')
 runtime_agent_device=st.radio("Choose runtime agent device combo",("TFLite-EdgeTPU","N2X-ORCA","N2X-ORCA1","OpenVINO-CPU"),index=2)
 runtime_agent,device=runtime_agent_device.split('-')[0],runtime_agent_device.split('-')[1]
 model_options=zoo.list_models(device=device,runtime=runtime_agent)
+activation_option=st.radio( 'Select activation function', ['SiLU', 'RELU6'])
+dataset_option=st.radio( 'Select a dataset option', ['COCO', 'Face','License Plate','Car','Hand'])
 st.header('Choose and Run a Model')
 st.text('Select a model and upload an image. Then click on the submit button')
 with st.form("model_form"):
-    model_name=st.selectbox("Choose a Model from the list", model_options)
+    act_filtered_model_options = [model if activation in model for model in model_options]
+    dataset_filtered_model_options=[model if dataset_option in model for model in act_filtered_model_options]
+    st.write('Number of models found = ', len(dataset_filtered_model_options))
+    model_name=st.selectbox("Choose a Model from the list", dataset_filtered_model_options)
     uploaded_file=st.file_uploader('input image')
     submitted = st.form_submit_button("Submit")
     if submitted: