Spaces:
Sleeping
Sleeping
pickle save directiory changed
Browse files- api/web_server.py +1 -1
- 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('
|
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()
|