.

pages/3_Verify.py

@@ -44,10 +44,10 @@ def select_document():
     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'
+    selection_image_left = MEDIA_ROOT+selection_left
     left.image(selection_image_left, use_column_width='always')
 
-    selection_image_right = MEDIA_ROOT+selection_right+'.png'
+    selection_image_right = MEDIA_ROOT+selection_right
     right.image(selection_image_right, use_column_width='always')
     return selection_image_left, selection_image_right