added query rotation

app.py

@@ -449,6 +449,9 @@ def search_and_retrieve(
         tfm.Resize((MATCHING_IMG_SIZE, MATCHING_IMG_SIZE), antialias=True)(query_image)
     )
 
+    
+
+
     progress(0.4, desc="Processing candidates")
     for faiss_idx in flat_indices:
 
@@ -456,6 +459,7 @@ def search_and_retrieve(
 
     
         best_rotation_index = faiss_idx // num_db_images
+        query_tensor = F.rotate(query_tensor, [0, 90, 180, 270][best_rotation_index] ) * -1
 
         if image_index in processed_image_indices:
             continue
@@ -513,9 +517,7 @@ def search_and_retrieve(
         candidate_img_tensor = tfm.Resize(
             (MATCHING_IMG_SIZE * 3, MATCHING_IMG_SIZE * 3), antialias=True
         )(candidate_img_tensor)
-        candidate_img_tensor = F.rotate(
-            candidate_img_tensor, [0, 90, 180, 270][best_rotation_index]
-        )
+    
 
         candidate_img_tensor = candidate_img_tensor.to(DEVICE)