Update Fsg_pp.py

Fsg_pp.py

@@ -25,16 +25,9 @@ class ImageGallery:
 
     def return_images(self, image_locs):
         self.imgz = image_locs
-
-        total_images = len(glob.glob('./Images/*'))
-        if total_images >= 20:
-            os.system("rm -r ./Images")
-            os.makedirs("./Images")
-
-        # Log the return value
-        logging.info(self.imgz)
         return self.imgz if self.imgz else []
 
+
     def get_select_index(self, evt: gr.SelectData):
         self.selected = evt.index
         return self.selected
@@ -85,6 +78,15 @@ def create_gallery_tab(tab_name, search_fn, search_inputs, gallery_instance, fn_
     with gr.Column():
         gallery=gr.Gallery(label="Image Preview", preview=True, object_fit="cover", container=True, columns=5)
 
+        total_images = len(glob.glob('./Images/*'))
+        if total_images >= 20:
+            os.system("rm -r ./Images")
+            os.makedirs("./Images")
+
+        # Log the return value
+        logging.info(gallery_instance.imgz)
+        
+
         with gr.Row():
             crop_btn = gr.Button(value="Crop Selected Image",variant='secondary')
             crop_btn.click(fn=fn_on_click, outputs=[image,tabs])