Update app.py

app.py

@@ -15,10 +15,12 @@ dataset_option=st.radio( 'Select a dataset option', ['COCO', 'Face','License Pla
 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"):
-    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)
+    filtered_model_list=[]
+    for model in model_options:
+        if activation_option in model and dataset_option in model:
+            filtered_model_list.append(model)
+    st.write('Number of models found = ', len(filtered_model_list))
+    model_name=st.selectbox("Choose a Model from the list", filtered_model_list)
     uploaded_file=st.file_uploader('input image')
     submitted = st.form_submit_button("Submit")
     if submitted: