krstakis commited on
Commit
290b57e
·
1 Parent(s): 08ca379

pickle save directiory changed

Browse files
Files changed (2) hide show
  1. api/web_server.py +1 -1
  2. core/initialization.py +1 -1
api/web_server.py CHANGED
@@ -11,7 +11,7 @@ class Query(BaseModel):
11
 
12
  app = FastAPI()
13
 
14
- with open('./engine.pickle', 'rb') as file:
15
  serialized_engine = file.read()
16
 
17
  prompt_search_engine = dill.loads(serialized_engine)
 
11
 
12
  app = FastAPI()
13
 
14
+ with open('core/engine.pickle', 'rb') as file:
15
  serialized_engine = file.read()
16
 
17
  prompt_search_engine = dill.loads(serialized_engine)
core/initialization.py CHANGED
@@ -15,7 +15,7 @@ def run():
15
 
16
  serialized_engine = dill.dumps(engine)
17
 
18
- with open("engine.pickle", "wb") as file:
19
  file.write(serialized_engine)
20
 
21
  run()
 
15
 
16
  serialized_engine = dill.dumps(engine)
17
 
18
+ with open("core/engine.pickle", "wb") as file:
19
  file.write(serialized_engine)
20
 
21
  run()