g8a9 commited on
Commit
daa8529
·
1 Parent(s): d00235e
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ def download_images():
39
  print("Done.")
40
 
41
 
42
- @st.cache
43
  def get_image_features(model, image_dir):
44
  image_size = model.config.vision_config.image_size
45
 
 
39
  print("Done.")
40
 
41
 
42
+ @st.cache(allow_output_mutation=True)
43
  def get_image_features(model, image_dir):
44
  image_size = model.config.vision_config.image_size
45