Updated main.yml

Author: anibalrojosan

.github/workflows/main.yml

@@ -23,13 +23,13 @@ jobs:
     - name: Install Python dependencies for model training
       run: |
         pip install --upgrade pip
-        pip install pandas scikit-learn joblib
+        pip install -r requirements.txt
 
     - name: Create models directory
       run: mkdir -p models
 
     - name: Train the model
-      run: python src/model.py
+      run: python -m src.model_training
 
     - name: Set up Docker Buildx
       uses: docker/setup-buildx-action@v3