dengkane commited on
Commit
800a230
1 Parent(s): cb8ce3f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -36,9 +36,6 @@ with open(knowledge_file, 'r', encoding='utf-8') as file:
36
  for line in file:
37
  knowledge.append(line.strip())
38
 
39
- # Create an index
40
- index = faiss.IndexFlatIP(300) # Use Inner Product (IP) as similarity measure
41
-
42
  # Perform embedding for the knowledge texts and add to index
43
  embeddings = []
44
  for text in knowledge:
@@ -46,6 +43,10 @@ for text in knowledge:
46
  embedding = get_embedding(text)
47
  embeddings.append(embedding)
48
  embeddings = np.array(embeddings)
 
 
 
 
49
  index.add(embeddings)
50
 
51
  # Get user input for a question
 
36
  for line in file:
37
  knowledge.append(line.strip())
38
 
 
 
 
39
  # Perform embedding for the knowledge texts and add to index
40
  embeddings = []
41
  for text in knowledge:
 
43
  embedding = get_embedding(text)
44
  embeddings.append(embedding)
45
  embeddings = np.array(embeddings)
46
+
47
+ # Create an index
48
+ index = faiss.IndexFlatIP(embeddings.shape[1]) # Use Inner Product (IP) as similarity measure
49
+
50
  index.add(embeddings)
51
 
52
  # Get user input for a question