Update app.py

app.py

@@ -14,7 +14,7 @@ HF_TOKEN = os.environ.get("HF_TOKEN") if os.environ.get("HF_TOKEN") else None #
 
 
 nb_req_simult=80 ########
-max_pending=8
+max_pending=4
 nb_gallery_model=5
 
 
@@ -517,7 +517,7 @@ def fonc_start(id_session,id_module,s,cont,list_models_to_gen):
                 
     if model_actu == "":        
         for model_plus_tasks in cache_list_task[f"{id_session}"]:
-            if model_plus_tasks["id_module"]==id_module:
+            if model_plus_tasks["pending_task"]==0:
                 try:
                     task_actu=model_plus_tasks["task"].pop()
                 except:
@@ -525,26 +525,14 @@ def fonc_start(id_session,id_module,s,cont,list_models_to_gen):
                 model_actu=model_plus_tasks["model"]
                 model_plus_tasks["pending_task"]+=1
                 print(f"find model : {model_actu}\n")
-                break
-            
-    if model_actu=="":
-        for model_plus_tasks in cache_list_task[f"{id_session}"]:
-            if model_plus_tasks["id_module"]==-1 :
-                if model_actu=="":
-                    
-                    try:
-                        task_actu=model_plus_tasks["task"].pop()
-                    except:
-                        continue
-                    model_actu=model_plus_tasks["model"]
-                    model_plus_tasks["pending_task"]+=1
-                    model_plus_tasks["id_module"]=id_module
-                    print(f"module num {id_module} take {model_actu}\n")
+                if len(model_plus_tasks["task"])==cache_text_actu[f"{id_session}"]["nb_tasks_by_model"]-1:
                     i=0
                     for model in list_models_to_gen:
                         if model_actu==model:
                             cache_text_actu[f"{id_session}"]['progress'][i]=1
                         i+=1
+                break
+            
                     
     if model_actu=="":
         cache_text_actu[f"{id_session}"]["nb_modules_use"]-=1