Updated lines 488 and 619 with: iterations=[0, 2, 8, 10]

app.py

@@ -485,7 +485,7 @@ class GWO:
         #plt.close()  # Close the figure to free up memory
         #return Image.open(buf)
 
-    def plot_positions(self, iterations=[0, 25, 75, 190]):
+    def plot_positions(self, iterations=[0, 2, 8, 10]):
         """Plot the positions of the particles over the specified iterations"""
         # Load the all_positions data from the .npy file
         all_positions = np.load('all_positions.npy', allow_pickle=True)
@@ -616,7 +616,7 @@ def optimize(npart, ndim, max_iter):
     dispersion_heatmap_plot = gwo.plot_dispersion_heatmap(x_range=(-6,6), y_range=(-6,6))
 
     # Plot the plot_positions 
-    plot_positions = gwo.plot_positions(iterations=[0,25, 75, 190])
+    plot_positions = gwo.plot_positions(iterations=[0, 2, 8, 10])
 
     # Plot the plot_3d_meshgrid_contour
     plot_3d_meshgrid_contour = gwo.plot_3d_meshgrid_contour(obj=obj)