gaurav22iiserb commited on
Commit
97ef835
·
verified ·
1 Parent(s): 4fe91a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -9,15 +9,15 @@ import numpy as np
9
  app = Flask(__name__)
10
 
11
  # Create a cache directory if it doesn't exist
12
- cache_dir = '/code/cache' # Or another suitable location
13
- os.makedirs(cache_dir, exist_ok=True)
14
 
15
 
16
 
17
  indexIVFPQ = faiss.read_index("indexIVFPQ .index")
18
 
19
  from sentence_transformers import SentenceTransformer
20
- model = SentenceTransformer('bert-base-nli-mean-tokens', cache_dir=cache_dir)
21
 
22
  import json
23
  with open('wiki_entries.json') as json_file:
 
9
  app = Flask(__name__)
10
 
11
  # Create a cache directory if it doesn't exist
12
+ #cache_dir = '/code/cache' # Or another suitable location
13
+ #os.makedirs(cache_dir, exist_ok=True)
14
 
15
 
16
 
17
  indexIVFPQ = faiss.read_index("indexIVFPQ .index")
18
 
19
  from sentence_transformers import SentenceTransformer
20
+ model = SentenceTransformer('bert-base-nli-mean-tokens')
21
 
22
  import json
23
  with open('wiki_entries.json') as json_file: