Spaces:
Sleeping
Sleeping
Commit ·
2cc7b15
1
Parent(s): a068dfc
added docker config
Browse files- Dockerfile +2 -0
Dockerfile
CHANGED
|
@@ -26,6 +26,8 @@ RUN mkdir -p /app/data /app/final_model /app/templates
|
|
| 26 |
|
| 27 |
# run the load_data_to_sqlite.py script to initialize the database
|
| 28 |
RUN python load_data_to_sqlite.py
|
|
|
|
|
|
|
| 29 |
|
| 30 |
# Expose port 7860 (HF Space requirement)
|
| 31 |
EXPOSE 7860
|
|
|
|
| 26 |
|
| 27 |
# run the load_data_to_sqlite.py script to initialize the database
|
| 28 |
RUN python load_data_to_sqlite.py
|
| 29 |
+
# Train the model during build (this persists across container restarts)
|
| 30 |
+
RUN python -c "from src.pipeline.training_pipeline import Trainingpipeline; Trainingpipeline().run_pipeline()"
|
| 31 |
|
| 32 |
# Expose port 7860 (HF Space requirement)
|
| 33 |
EXPOSE 7860
|