Update app.py

app.py

@@ -33,7 +33,16 @@ def create_chunks(text, chunk_size):
 
 # Function to create FAISS vector store
 def create_faiss_index(chunks, vector_dim):
-    index = faiss.IndexFlatL2(vector_dim)
+    # Check if GPU is available and use it
+    if faiss.get_num_gpus() > 0:
+        st.write("Using GPU for FAISS indexing.")
+        resource = faiss.StandardGpuResources()  # Initialize GPU resources
+        index_flat = faiss.IndexFlatL2(vector_dim)
+        index = faiss.index_cpu_to_gpu(resource, 0, index_flat)
+    else:
+        st.write("Using CPU for FAISS indexing.")
+        index = faiss.IndexFlatL2(vector_dim)
+
     embeddings = np.random.rand(len(chunks), vector_dim).astype('float32')  # Replace with real embeddings
     index.add(embeddings)
     return index, embeddings