Update app.py (#3)

- Update app.py (4a5059070f83519da16a68fa533795b291a6309b)


Co-authored-by: Marcelo Moreno <[email protected]>

app.py

@@ -4,6 +4,11 @@ import streamlit as st
 import pandas as pd
 
 
+@st.cache_data
+def load_example_file(file):
+    with open(file, "rb") as f:
+        return f.read()
+
 
 def main():
     # Load models
@@ -22,9 +27,10 @@ def main():
         "Example 2": "example2.zip",
         "Example 3": "example3.zip"
     }
+    
     for name, file in example_files.items():
-        with open(file, "rb") as f:
-            st.download_button(label=name, data=f, file_name=file, mime="application/zip")
+        data = load_example_file(file)
+        st.download_button(label=name, data=data, file_name=file, mime="application/zip")
     
 
     # File uploader