Spaces:
Running
Running
bugfix
Browse files
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 |
|