Update app.py

app.py

@@ -322,10 +322,11 @@ def reset_cache_image_actu(id,cache_image_actu=cache_image_actu):
 def reset_cache_image_all_sessions(cache_image=cache_image):
     cache_image.clear()
     return 
-def set_session(id_session,cache_image,cache_image_actu):
-    print(f"id = {id_session}")
-    cache_image[f"{id_session}"]=[]
-    cache_image_actu[f"{id_session}"]=[]
+def set_session(id,cache_image,cache_image_actu):
+    print(f"id = {id}")
+    cache_image[f"{id}"]=[]
+    cache_image_actu[f"{id}"]=[]
+    print(cache_image)
     return
 
 def disp_models(group_model_choice,nb_rep=nb_rep):
@@ -445,7 +446,7 @@ def make_me():
                         b22 = gr.Button('Reset Gallery All')
                     b11.click(load_gallery_actu,[gallery,id_session],gallery)
                     b12.click(load_gallery,[gallery,id_session],gallery)
-                    b21.click(reset_gallery,[gallery,id_session],gallery)
+                    b21.click(reset_gallery,[gallery],gallery)
                     b22.click(reset_cache_image,[id_session],gallery)
                     for m, o in zip(current_models_g, output_g):
                         #o.change(add_gallery, [o, m, gallery], [gallery])