Update app.py

app.py

@@ -28,13 +28,12 @@ data = data[data['embedding'].apply(lambda x: x.size > 0)]
 dimension = len(data['embedding'][0])
 res = faiss.StandardGpuResources()  # use a single GPU
 
-# Check available GPU devices
-num_gpus = faiss.get_num_gpus()
-if num_gpus > 0:
+# Create FAISS index
+if faiss.get_num_gpus() > 0:
     gpu_index = faiss.IndexFlatL2(dimension)
     gpu_index = faiss.index_cpu_to_gpu(res, 0, gpu_index)  # move to GPU
 else:
-    raise RuntimeError("No GPU devices available.")
+    gpu_index = faiss.IndexFlatL2(dimension)  # fall back to CPU
 
 gpu_index.add(np.stack(data['embedding'].values))