Update app.py

app.py

@@ -9,6 +9,11 @@ st.set_page_config(page_title='Smoke & Fire Detection')
 if 'history' not in st.session_state:
     st.session_state.history = []
 
+@st.cache()
+def demo_img():
+    demo_img = Image.open("./demo.jpg")
+    compute(demo_img)
+
 @st.cache(persist=True)
 def loadModel():
     pipeline = pip(task="image-classification", model="EdBianchi/vit-fire-detection")
@@ -49,8 +54,7 @@ if file_name is not None:
     compute(image)
 
 # DEMO IMAGE
-demo_img = Image.open("./demo.jpg")
-compute(demo_img)
+demo_img()
 
 # SIDEBAR
 #st.sidebar.write("""""")