Spaces:
Sleeping
Sleeping
fixed imports and changed dockerfile
Browse files- Dockerfile +3 -2
- api/web_server.py +1 -1
Dockerfile
CHANGED
@@ -30,11 +30,12 @@ RUN pip install -r requirements.txt
|
|
30 |
# Copy the API files
|
31 |
COPY ./api .
|
32 |
|
33 |
-
ENV PYTHONPATH="/app"
|
34 |
-
|
35 |
# Copy the serialized engine from the first stage
|
36 |
COPY --from=core /app/core/engine.pickle /app/api/engine.pickle
|
37 |
|
|
|
|
|
|
|
38 |
# Expose the API port
|
39 |
EXPOSE 9999
|
40 |
|
|
|
30 |
# Copy the API files
|
31 |
COPY ./api .
|
32 |
|
|
|
|
|
33 |
# Copy the serialized engine from the first stage
|
34 |
COPY --from=core /app/core/engine.pickle /app/api/engine.pickle
|
35 |
|
36 |
+
# Set the PYTHONPATH to include the /app directory
|
37 |
+
ENV PYTHONPATH="/app"
|
38 |
+
|
39 |
# Expose the API port
|
40 |
EXPOSE 9999
|
41 |
|
api/web_server.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
import dill
|
2 |
from fastapi import FastAPI, HTTPException
|
3 |
from pydantic import BaseModel
|
4 |
-
from
|
5 |
|
6 |
|
7 |
class Query(BaseModel):
|
|
|
1 |
import dill
|
2 |
from fastapi import FastAPI, HTTPException
|
3 |
from pydantic import BaseModel
|
4 |
+
from core.search_engine import PromptSearchEngine
|
5 |
|
6 |
|
7 |
class Query(BaseModel):
|