update

app.py

@@ -22,6 +22,12 @@ import config
 from utils import load_model, infer_uploaded_image, infer_uploaded_video, infer_uploaded_webcam
 import os
 
+query_params = st.experimental_get_query_params()
+
+qmodel = 'crowded_human'
+if 'model' in query_params:
+    qmodel = query_params['model'][0]
+
 # setting page layout
 st.set_page_config(
     page_title="YOLO.dog",
@@ -73,20 +79,24 @@ for i in range(0, model_count):
         #else:
         #    download_file_from_google_drive(gdrive_id, path_model)
 
-print('models_list', models_list)
+#print('models_list', models_list)
+if qmodel not in models_list:
+    qmodel = models_list[0]
+
 
 # main page heading
-st.title("Models Demo")
+st.title( model_info[qmodel] )
 
 # sidebar
-st.sidebar.header("DL Model Config")
+st.sidebar.header("Model Config")
 
 # model options
 task_type = "Detection"
 
 model_type = st.sidebar.selectbox(
-   "Model types",
-   tuple(models_list))
+   "Models list",
+   models_list,
+   index=models_list.index(qmodel) )
 
 confidence = float(st.sidebar.slider(
     "Select Model Confidence", 30, 100, 50)) / 100