Update app.py

app.py

@@ -12,11 +12,12 @@ def main():
         st.write('## Input')
         uploaded_file = st.file_uploader('Upload the image file (PNG or JPG)', type=['png', 'jpg'], help='help')
         input_file = st.file_uploader('Upload the input file (TXT)', type=['txt'], help='help')
+        ocre = st.selectbox('OCR Engine', ['Hive', 'Tesseract'])
 
     if st.button('Find Order ID') and uploaded_file and input_file:
         st.write('## Output')
         model = tf.keras.models.load_model('model.h5')
-        result = find_order_id(uploaded_file, input_file, model)
+        result = find_order_id(uploaded_file, input_file, model, ocre)
         if result['status'] == 'success':
             st.success(result['message'])
         elif result['status'] == 'warning':