Update app.py

app.py

@@ -56,7 +56,6 @@ def resize(img, target_res=224, resize=True, to_pil=True, edge=False, sampling_f
 # ─── Feature extraction ──────────────────────────────────────────
 @spaces.GPU
 def get_processed_features_dino(num_patches, img,use_dummy):
-    torch.cuda.empty_cache()
     batch = extractor_vit.preprocess_pil(img)
     features_dino = extractor_vit.extract_descriptors(batch.to(extractor_vit.device), layer=11, facet='token') \
                                     .permute(0,1,3,2) \
@@ -69,7 +68,9 @@ def get_processed_features_dino(num_patches, img,use_dummy):
             desc = aggre_net(features_dino)
         norms = torch.linalg.norm(desc, dim=1, keepdim=True)
         desc = desc / (norms + 1e-8)
-    return desc.cpu()  # shape [1, C, num_patches, num_patches]
+    desc = desc.cpu()
+    torch.cuda.empty_cache()
+    return desc  # shape [1, C, num_patches, num_patches]
 
 # ─── Similarity computation ───────────────────────────────────────
 def get_sim(