Update app.py

app.py

@@ -207,8 +207,14 @@ if uploaded_file is not None:
 
     if transformation:
         clear_folder()
-        per_trans = max(1,(image_count // len(transformation)))
-        trans_count=[per_trans+1 if (per_trans * len(transformation))!= image_count else per_trans][0]
+        # per_trans = max(1,(image_count // len(transformation)))
+        # trans_count=[per_trans+1 if (per_trans * len(transformation))!= image_count else per_trans][0]
+        num_trans = len(transformation)
+        base = image_count // num_trans
+        remainder = image_count % num_trans
+    
+    # Create a list like [base+1, base, base, ...] to ensure sum == image_count
+        counts = [base + 1 if i < remainder else base for i in range(num_trans)]
 
         if "Translation" in transformation:
             translation(img,trans_count)