Update pages/3_Adversarial_attack.py

pages/3_Adversarial_attack.py

@@ -118,13 +118,19 @@ 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]
-            epsilon_container.progress(0, text='Checking epsilon')
+            with epsilon_container:
+                epsilon_container_text = 'Checking epsilon'
+                st.write(epsilon_container_text)
+                st.progress(0)
         
             for i, e in enumerate(epsilons):
-                print(e)
                 
                 perturbed_data, new_prob, new_id, new_label = perform_attack(input_image, input_id-1, e)
-                epsilon_container.progress(i/len(epsilons), text=f'Checking epsilon={e:.3f}. Confidence={new_prob*100:.1f}%')
+                with epsilon_container:
+                    epsilon_container_text = f'Checking epsilon={e:.3f}. Confidence={new_prob*100:.1f}%'
+                    st.write(epsilon_container_text)
+                    st.progress(i/len(epsilons))
+                
                 epsilon = e
 
                 if new_id != input_id - 1: