fixing tensor device

app.py

@@ -12,7 +12,7 @@ from collections import Counter
 from SimSearch import FaissCosineNeighbors
 
 # HELPERS
-to_np = lambda x: x.data.to('cpu').numpy()
+# to_np = lambda x: x.data.to('cpu').numpy()
 
 # DOWNLOAD THE DATASET and Files
 
@@ -54,7 +54,7 @@ class GamePhysicsSearcher:
     y_train = []
 
     for i, vfile in enumerate(tqdm(self.all_embeddings)):
-      vfeatures = to_np(self.read_features(vfile))
+      vfeatures = self.read_features(vfile)
       features[vfile.split('/')[-1]] = vfeatures
       X_train.extend(vfeatures)
       y_train.extend([i]*vfeatures.shape[0])
@@ -74,7 +74,7 @@ class GamePhysicsSearcher:
     with torch.no_grad():
       text_features = self.CLIP_MODEL.encode_text(text_tokens).float()
     text_features /= text_features.norm(dim=-1, keepdim=True)
-    return to_np(text_features)
+    return text_features
 
   # Source: https://stackoverflow.com/a/480227
   def f7(self, seq):