Update app.py

app.py

@@ -23,6 +23,14 @@ confidence = float(st.sidebar.slider(
 source_img = None
 
 if model_type == 'Detection — **POT-YOLO**':
+    st.sidebar.header("Image/Video Config")
+
+    source_radio = st.sidebar.radio(
+        "Select Source", settings.SOURCES_LIST)
+
+    if source_radio == settings.IMAGE:
+        source_img = st.sidebar.file_uploader(
+            "Choose an image...", type=("jpg", "jpeg", "png", 'bmp', 'webp'))
     model_path = Path(settings.DETECTION_MODEL_1)
     st.title("Object Detection And Tracking using POT-YOLO")
     try:
@@ -71,6 +79,14 @@ if model_type == 'Detection — **POT-YOLO**':
                     # st.write(ex)
                     st.write("No image is uploaded yet!")
 elif model_type == 'Detection — **African Wildlife** (BETA)':
+    st.sidebar.header("Image/Video Config")
+
+    source_radio = st.sidebar.radio(
+        "Select Source", settings.SOURCES_LIST)
+
+    if source_radio == settings.IMAGE:
+        source_img = st.sidebar.file_uploader(
+            "Choose an image...", type=("jpg", "jpeg", "png", 'bmp', 'webp'))
     model_path = Path(settings.DETECTION_MODEL_2)
     st.title("Object Detection And Tracking using African-YOLO")
     try:
@@ -119,16 +135,9 @@ elif model_type == 'Detection — **African Wildlife** (BETA)':
                     # st.write(ex)
                     st.write("No image is uploaded yet!")
 
-
-st.sidebar.header("Image/Video Config")
-source_radio = st.sidebar.radio(
-    "Select Source", settings.SOURCES_LIST)
-
-
 if source_radio == settings.IMAGE:
-    source_img = st.sidebar.file_uploader(
-        "Choose an image...", type=("jpg", "jpeg", "png", 'bmp', 'webp'))
-
+    print('Loaded')
+    # Image source selected
 
 elif source_radio == settings.VIDEO:
     helper.play_stored_video(confidence, model)