Update server_medical_RAM_optimize.py

server_medical_RAM_optimize.py

@@ -72,7 +72,7 @@ class ImageSearchEngine:
         if isinstance(text, str):
             text = [text]
         
-        tokens = self.tokenizer(text).to(device)
+        tokens = self.tokenizer(text).to(self.device)
         
         with torch.no_grad():
             text_features = self.model.encode_text(tokens)    
@@ -81,7 +81,7 @@ class ImageSearchEngine:
         return text_features.cpu().numpy()
 
     def image_to_vector(self, image):
-        image_tensor = self.preprocess(image).unsqueeze(0).to(device)
+        image_tensor = self.preprocess(image).unsqueeze(0).to(self.device)
         
         with torch.no_grad():
             image_features = self.model.encode_image(image_tensor)