Update app file

app.py

@@ -23,11 +23,24 @@ st.title('P&ID Object Detection')
 st.subheader(' Identify valves and pumps with deep learning model ', divider='rainbow')
 st.caption('Developed by Deep Drawings Co.')
 
+@st.cache_data(show_spinner=False)
+def download_comparison_images():
+    sahi.utils.file.download_from_url(
+        'https://d1afc1j4569hs1.cloudfront.net/plant-pid.png',
+        'plant_pid.png',
+    )
+    sahi.utils.file.download_from_url(
+        'https://d1afc1j4569hs1.cloudfront.net/prediction_visual.png',
+        'prediction_visual.png',
+    )
+
+download_comparison_images()
+
 if "output_1" not in st.session_state:
-    st.session_state["output_1"] = sahi.utils.cv.read_image_as_pil(IMAGE_TO_URL['plant_pid.png'])
+    st.session_state["output_1"] = Image.open('plant_pid.png')
 
 if "output_2" not in st.session_state:
-    st.session_state["output_2"] = sahi.utils.cv.read_image_as_pil(IMAGE_TO_URL['prediction_visual.png'])
+    st.session_state["output_2"] = Image.open('prediction_visual.png')
 
 col1, col2, col3 = st.columns(3, gap='medium')
 with col1: