Fix for streamlit 1.17.0

app.py

@@ -23,15 +23,25 @@ from paddleocr import PaddleOCR
 import postprocess
 
 
-@st.cache_resource(ttl=3600)
-def load_models():
+@st.experimental_singleton(ttl=3600)
+def load_ocr_instance():
     ocr_instance = PaddleOCR(use_angle_cls=False, lang='en', use_gpu=True)
+    return ocr_instance
+
+
+@st.experimental_singleton(ttl=3600)
+def load_detection_model():
     detection_model = torch.hub.load('ultralytics/yolov5', 'custom', 'weights/detection_wts.pt', force_reload=True, skip_validation=True, trust_repo=True)
+    return detection_model
+
+
+@st.experimental_singleton(ttl=3600)
+def load_structure_model():
     structure_model = torch.hub.load('ultralytics/yolov5', 'custom', 'weights/structure_wts.pt', force_reload=True, skip_validation=True, trust_repo=True)
-    return ocr_instance, detection_model, structure_model
+    return structure_model
 
 
-ocr_instance, detection_model, structure_model = load_models()
+ocr_instance, detection_model, structure_model = load_ocr_instance(), load_detection_model(), load_structure_model()
 
 detection_class_names = ['table', 'table rotated', 'no object']
 structure_class_names = [

requirements.txt

@@ -74,7 +74,6 @@ setuptools>=65.5.1 # Snyk vulnerability fix
 # ultralytics  # HUB https://hub.ultralytics.com
 
 # Other
-streamlit>=1.21.0
 pytesseract==0.3.10
 # beautifulsoup4==4.11.1
 xlsxwriter