Update app.py

app.py

@@ -28,9 +28,9 @@ import numpy as np
 import tempfile
 from pathlib import Path
 
-# Update the numpy version check at the top of the file
-if not hasattr(np, '__version__') or tuple(map(int, np.__version__.split('.'))) > (1, 24, 0):
-    print(f"Warning: Using numpy version {np.__version__}. Some features may not work properly.")
+# Update the numpy version check
+if not hasattr(np, '__version__') or tuple(map(int, np.__version__.split('.'))) != (1, 23, 5):
+    print(f"Warning: Using numpy version {np.__version__}. Expected version 1.23.5")
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -104,9 +104,16 @@ def load_sentence_transformer():
 # Load FAISS index
 def load_faiss_index():
     print("Loading FAISS index...")
-    index = faiss.read_index(faiss_index_path)
-    print("FAISS index loaded successfully.")
-    return index
+    try:
+        index = faiss.read_index(faiss_index_path)
+        # Ensure the index is on CPU
+        if hasattr(faiss, 'StandardGpuResources'):
+            index = faiss.index_gpu_to_cpu(index)
+        print("FAISS index loaded successfully.")
+        return index
+    except Exception as e:
+        print(f"Error loading FAISS index: {str(e)}")
+        raise
 
 # Load dataset
 def load_dataset():