Update backend/adversarial_attack.py

backend/adversarial_attack.py

@@ -89,7 +89,8 @@ def find_smallest_epsilon(input_image, target):
             return perturbed_data, new_prob, new_id, new_label, epsilon
     return None
 
-@st.cache_data
+# @st.cache_data
+@st.cache(allow_output_mutation=True)
 def generate_images(image_id, epsilon=0):
     model, feature_extractor = load_model("ConvNeXt")
     original_image_dict = load_image(image_id)