read initial comparison iamges from cache

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import sahi.utils.file
 import sahi.utils.mmdet
 import sahi.model
 from PIL import Image
@@ -19,7 +20,19 @@ IMAGE_TO_URL = {
 
 
 @st.cache(allow_output_mutation=True, show_spinner=False)
-def get_model(model_name: str):
+def download_comparison_images():
+    sahi.utils.file.download_from_url(
+        "https://user-images.githubusercontent.com/34196005/143309873-c0c1f31c-c42e-4a36-834e-da0a2336bb19.jpg",
+        "highway2-yolox.jpg",
+    )
+    sahi.utils.file.download_from_url(
+        "https://user-images.githubusercontent.com/34196005/143309867-42841f5a-9181-4d22-b570-65f90f2da231.jpg",
+        "highway2-sahi.jpg",
+    )
+
+
+@st.cache(allow_output_mutation=True, show_spinner=False)
+def get_model():
     model_path = "yolox.pt"
     sahi.utils.file.download_from_url(
         MMDET_YOLOX_MODEL_URL,
@@ -73,18 +86,16 @@ st.set_page_config(
     initial_sidebar_state="auto",
 )
 
+download_comparison_images()
+
 if "last_spinner_texts" not in st.session_state:
     st.session_state["last_spinner_texts"] = SpinnerTexts()
 
 if "output_1" not in st.session_state:
-    st.session_state["output_1"] = sahi.utils.cv.read_image_as_pil(
-        IMAGE_TO_URL["highway2-yolox.jpg"]
-    )
+    st.session_state["output_1"] = Image.open("highway2-yolox.jpg")
 
 if "output_2" not in st.session_state:
-    st.session_state["output_2"] = sahi.utils.cv.read_image_as_pil(
-        IMAGE_TO_URL["highway2-sahi.jpg"]
-    )
+    st.session_state["output_2"] = Image.open("highway2-sahi.jpg")
 
 st.markdown(
     """