removing the gradients in the model evaluation

search.py

@@ -39,7 +39,9 @@ def process_text_for_encoder(text, model):
     encoded_input = tokenizer(text, return_tensors='pt', padding='max_length', truncation=True, max_length=256)
     input_ids = encoded_input['input_ids']
     attention_mask = encoded_input['attention_mask']
-    output = model(input_ids=input_ids, attention_mask=attention_mask)
+    model.eval()
+    with torch.no_grad():
+        output = model(input_ids=input_ids, attention_mask=attention_mask)
     return output.detach().numpy().tolist()[0]
 
 def process_image_for_encoder(image, model):
@@ -50,7 +52,9 @@ def process_image_for_encoder(image, model):
             return_tensors="pt", 
             do_resize=True
             )['pixel_values']
-    output =  model(pixel_values=image_tensor)
+    model.eval()
+    with torch.no_grad():
+        output =  model(pixel_values=image_tensor)
     return output.detach().numpy().tolist()[0]
 
 def search_similarity(input, mode, top_k=5):