Update app.py

app.py

@@ -19,10 +19,12 @@ def init_session_states():
 
 
 
-init_session_states()
-import easyocr
+
+
 
 def main():
+    init_session_states()
+    import easyocr
     st.session_state['disp'].text("Env setup up Complete")
     uploaded_file = st.file_uploader("Choose image file to detect text",type=['jpeg','jpg'])
     if uploaded_file is not None:
@@ -30,9 +32,11 @@ def main():
         st.write(file_details)
         image = load_image(uploaded_file)
         st.image(image,width=250)
+        st.session_state['disp'].text("Detecting text....")
         reader = easyocr.Reader(['en'],gpu=True)
         bound = reader.readtext(image)
         st.write(str(bound))
+        st.session_state['disp'].text("Detection complete")