Update app.py

app.py

@@ -21,23 +21,21 @@ def split_models(models,nb_models):
         models_lis_temp.append(models_temp)
     return models_lis_temp
 
-def split_models_4x10(models,nb_models):
+def split_models_axb(models,a,b):
     models_temp=[]
     models_lis_temp=[]
     i=0
-    nb_models_4x10=10
+    nb_models=b
     for m in models:
-        models_temp.append(m)
-        models_temp.append(m)
-        models_temp.append(m)
-        models_temp.append(m)
+        for j in range(a):
+            models_temp.append(m)
         i=i+1
-        if i%nb_models_4x10==0:
+        if i%nb_models==0:
             models_lis_temp.append(models_temp)
             models_temp=[]
     if len(models_temp)>1:
         models_lis_temp.append(models_temp)
-    return models_lis_temp , 40
+    return models_lis_temp , a*b
 
 def split_models_8x3(models,nb_models):
     models_temp=[]
@@ -56,7 +54,7 @@ def split_models_8x3(models,nb_models):
 
 """models_test=split_models_x3(models,nb_models)"""
 """models_test=split_models(models,nb_models)"""
-models_test , nb_models =split_models_4x10(models,nb_models)
+models_test , nb_models =split_models_axb(models,2,20)
 
 def get_current_time():
     now = datetime.now()