Spaces:
Sleeping
Sleeping
fixing searcg_enigne problem
Browse files- Dockerfile +1 -0
- api/web_server.py +1 -1
Dockerfile
CHANGED
@@ -31,6 +31,7 @@ COPY ./api ./api
|
|
31 |
|
32 |
# Copy the serialized engine from the first stage
|
33 |
COPY --from=core /app/core/engine.pickle /app/api/engine.pickle
|
|
|
34 |
|
35 |
# Set the PYTHONPATH to include the /app directory
|
36 |
ENV PYTHONPATH="/app"
|
|
|
31 |
|
32 |
# Copy the serialized engine from the first stage
|
33 |
COPY --from=core /app/core/engine.pickle /app/api/engine.pickle
|
34 |
+
COPY --from=core /app/core/search_engine /app/api/search_engine
|
35 |
|
36 |
# Set the PYTHONPATH to include the /app directory
|
37 |
ENV PYTHONPATH="/app"
|
api/web_server.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
import dill
|
2 |
from fastapi import FastAPI, HTTPException
|
3 |
from pydantic import BaseModel
|
4 |
-
|
5 |
|
6 |
|
7 |
class Query(BaseModel):
|
|
|
1 |
import dill
|
2 |
from fastapi import FastAPI, HTTPException
|
3 |
from pydantic import BaseModel
|
4 |
+
from search_engine import PromptSearchEngine
|
5 |
|
6 |
|
7 |
class Query(BaseModel):
|