clearing cache

main.py

@@ -58,7 +58,14 @@ def get_masks():
         if image is None:
             raise ValueError("Image not found or unable to read.")
         
+        if cudaOrNah == "cuda":
+            torch.cuda.empty_cache()
+        
         masks = mask_generator.generate(image)
+
+        if cudaOrNah == "cuda":
+            torch.cuda.empty_cache()
+
         masks = sorted(masks, key=(lambda x: x['area']), reverse=True)
 
         def is_background(segmentation):