Update app.py

app.py

@@ -97,7 +97,9 @@ def generate_image_embeddings(zip_file):
                 normalized_features = image_features / image_features.norm(
                     dim=-1, keepdim=True
                 )
-                embeddings.append(normalized_features.cpu().numpy())
+                embeddings.append(
+                    normalized_features.to(dtype=torch.float32).cpu().numpy()
+                )
 
         embeddings = np.vstack(embeddings)
         print(f"Embeddings shape: {embeddings.shape}")
@@ -202,7 +204,9 @@ def generate_video_embeddings(video_path, fps):
                 normalized_features = image_features / image_features.norm(
                     dim=-1, keepdim=True
                 )
-                embeddings.append(normalized_features.cpu().numpy())
+                embeddings.append(
+                    normalized_features.to(dtype=torch.float32).cpu().numpy()
+                )
 
         embeddings = np.vstack(embeddings)
         print(f"Embeddings shape: {embeddings.shape}")