YALCINKAYA commited on
Commit
3fdeaa6
·
verified ·
1 Parent(s): 8434101

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -73,10 +73,11 @@ def initialize_faiss():
73
  index = faiss.IndexIDMap(faiss.IndexFlatL2(384))
74
 
75
  # Move to GPU if available
76
- if torch.cuda.is_available():
77
- print("CUDA is available, moving FAISS index to GPU.")
78
- index = faiss.index_cpu_to_all_gpus(index)
79
- print("FAISS index is now on GPU.")
 
80
 
81
  return index
82
 
 
73
  index = faiss.IndexIDMap(faiss.IndexFlatL2(384))
74
 
75
  # Move to GPU if available
76
+ # if torch.cuda.is_available():
77
+ # print("CUDA is available, moving FAISS index to GPU.")
78
+ # index = faiss.index_cpu_to_all_gpus(index)
79
+
80
+ # print("FAISS index is now on GPU.")
81
 
82
  return index
83