Update app.py

app.py

@@ -331,11 +331,10 @@ def add_gallery(image, model_str, gallery):
 
 def reset_gallery(gallery):
     return add_gallery(None,"",[])
-
-def load_gallery(gallery,id):
+def fonc_load_gallery(id_session,gallery):
     gallery = reset_gallery(gallery)
-    for c in cache_image[f"{id}"]:
-        gallery=add_gallery(c[0],c[1],gallery)
+    for i in range(len(cache_image[f"{id_session}"])):
+        gallery=add_gallery(cache_image[f"{id_session}"][i],cache_image[f"{id_session}"][i]["model"],gallery)
     return gallery
 
 def fonc_start(id_session,id_module):
@@ -460,11 +459,7 @@ def make_me():
                     button_load_gallery=("Load Gallery")
                     button_load_gallery.click(fonc_load_gallery,[id_session,gallery],[])
                     
-def fonc_load_gallery(id_session,gallery):
-    gallery = reset_gallery(gallery)
-    for i in range(len(cache_image[f"{id_session}"])):
-        gallery=add_gallery(cache_image[f"{id_session}"][i],cache_image[f"{id_session}"][i]["model"],gallery)
-    return gallery
+