.

pages/3_Verify.py

@@ -40,8 +40,8 @@ def select_document():
     left, right = st.columns(2) # Create two columns
     # dropdown box in left column
     files = ["01_039_merged","01_068_merged", "02_042", "02_068","06_018_merged", "01_042_merged", "02_039", "02_043_merged", "03_018", "10_043"]
-    selection_left = str(left.selectbox('Select document to run inference 1', files))
-    selection_right = str(right.selectbox('Select document to run inference 2', files))
+    selection_left = str(left.selectbox('Select document to run inference 1', os.listdir(MEDIA_ROOT)))
+    selection_right = str(right.selectbox('Select document to run inference 2', os.listdir(MEDIA_ROOT)))
     
     # select corresponding document image from media/documents
     selection_image_left = MEDIA_ROOT+selection_left+'.png'