Update pages/3_Adversarial_attack.py

pages/3_Adversarial_attack.py

@@ -118,7 +118,7 @@ if pred_class_id == (input_id-1) and (epsilon_button or find_epsilon or setting_
             perturbed_data, new_prob, new_id, new_label = perform_attack(input_image, input_id-1, epsilon)
         else:
             epsilons = [i*step_epsilon for i in range(1, 1001) if i*step_epsilon <= max_epsilon]
-            with epsilon_container:
+            with epsilon_container.container():
                 epsilon_container_text = 'Checking epsilon'
                 st.write(epsilon_container_text)
                 st.progress(0)
@@ -126,7 +126,7 @@ if pred_class_id == (input_id-1) and (epsilon_button or find_epsilon or setting_
             for i, e in enumerate(epsilons):
                 
                 perturbed_data, new_prob, new_id, new_label = perform_attack(input_image, input_id-1, e)
-                with epsilon_container:
+                with epsilon_container.container():
                     epsilon_container_text = f'Checking epsilon={e:.3f}. Confidence={new_prob*100:.1f}%'
                     st.write(epsilon_container_text)
                     st.progress(i/len(epsilons))