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):
|