Allow clearing of cache

app.py

@@ -4,7 +4,7 @@ import pandas as pd
 import json
 import os
 import numpy as np
-from streamlit.elements import markdown
+from streamlit import caching
 from PIL import Image
 from model.flax_clip_vision_mbart.modeling_clip_vision_mbart import (
     FlaxCLIPVisionMBartForConditionalGeneration,
@@ -110,9 +110,9 @@ if state.image_file is None:
     image = plt.imread(image_path)
     state.image = image
 
-# col1, col2 = st.beta_columns([6, 4])
+col1, col2 = st.beta_columns([5, 5])
 
-if st.button("Get a random example", help="Get a random example from one of the seeded examples."):
+if col1.button("Get a random example", help="Get a random example from one of the seeded examples."):
     sample = dummy_data.sample(1).reset_index()
     state.image_file = sample.loc[0, "image_file"]
     state.caption = sample.loc[0, "caption"].strip("- ")
@@ -122,7 +122,8 @@ if st.button("Get a random example", help="Get a random example from one of the
     image = plt.imread(image_path)
     state.image = image
 
-# col2.write("OR")
+if col2.sidebar.button("Clear All Cache"):
+    caching.clear_cache()
 
 # uploaded_file = col2.file_uploader("Upload your image", type=["png", "jpg", "jpeg"])
 # if uploaded_file is not None: