Kshitijk20 commited on
Commit
2cc7b15
·
1 Parent(s): a068dfc

added docker config

Browse files
Files changed (1) hide show
  1. 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