Update pipeline.py

pipeline.py

@@ -2,7 +2,6 @@ from typing import List
 import torch
 from transformers import SamModel, SamProcessor
 from PIL import Image
-from io import BytesIO
 import numpy as np
 
 
@@ -17,7 +16,7 @@ class PreTrainedPipeline():
         self.model.eval()
         self.model = self.model.to(self.device)
 
-    def __call__(self, inputs: "Image.Image") -> BytesIO:
+    def __call__(self, inputs: "Image.Image") -> List[float]:
         raw_image = inputs.convert("RGB")
         inputs = self.processor(raw_image, return_tensors="pt").to(self.device)
         feature_vector = self.model.get_image_embeddings(