VGreatVig07 commited on
Commit
c445aa7
·
verified ·
1 Parent(s): 5e1e666

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ gemini_model = genai.GenerativeModel("gemini-2.0-flash")
15
  # Load vector index and dataframe
16
  index = faiss.read_index("shl_vector_index.faiss")
17
  df = pd.read_csv("shl_combined_assessments.csv")
18
- model = SentenceTransformer("all-MiniLM-L6-v2")
19
 
20
  # Helper to format the result row
21
  def format_row(row):
 
15
  # Load vector index and dataframe
16
  index = faiss.read_index("shl_vector_index.faiss")
17
  df = pd.read_csv("shl_combined_assessments.csv")
18
+ model = SentenceTransformer("all-MiniLM-L6-v2", device="cpu")
19
 
20
  # Helper to format the result row
21
  def format_row(row):