Add the cache decorator to the function and the newly required variab…

…les.
hreikin · Mar 9, 2023 · fb48218 · fb48218
1 parent 6555495
commit fb48218
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/streamlit_uploads_library/gallery.py b/src/streamlit_uploads_library/gallery.py
@@ -9,5 +9,6 @@ def __init__(self, directory, file_extensions=(".png", ".jpg", ".jpeg"), number_
     def fetch_files(self):
         return super().fetch_files()
 
+    @st.cache_resource(experimental_allow_widgets=True, show_spinner="Refreshing gallery...")
     def create_gallery(_self):
-        return super().create_library()
+        return super().create_library(_self.number_of_columns, _self.show_details)