fix: release GPU tensors after prediction (#144)

---------

Co-authored-by: Hilmar Lapp <[email protected]>

Author: mitchellxh

src/bioclip/predict.py

@@ -278,6 +278,8 @@ def create_probabilities_for_images(self, images: List[str] | List[PIL.Image.Ima
         images = [self.ensure_rgb_image(image) for image in images]
         img_features = self.create_image_features(images)
         probs = self.create_probabilities(img_features, txt_features)
+        # prevent accumulation of GPU VRAM
+        probs = probs.detach().cpu()
         result = {}
         for i, key in enumerate(keys):
             result[key] = probs[i]