git action added

requirements updated

.github/{main.yaml → main.yml}

@@ -1,21 +1,20 @@
-name: Sync to Hugging Face Hub
+name: Sync to Hugging Face hub
 
 on:
   push:
     branches: [main]
+
+  # to run this workflow manually from the Actions tab
   workflow_dispatch:
 
 jobs:
   sync-to-hub:
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v3
+      - uses: actions/checkout@v2
         with:
           fetch-depth: 0
-      - name: Push to Hugging Face Hub
+      - name: Push to hub
         env:
           HF_TOKEN: ${{ secrets.HF_TOKEN }}
-        run: |
-          git config --global user.email "[email protected]"
-          git config --global user.name "HAKIM-ML"
-          git push --force https://Md-Hakim:${{ secrets.HF_TOKEN }}@huggingface.co/spaces/Md-Hakim/image-colorization-mlops main
+        run: git push --force https://Md-Hakim:[email protected]/spaces/Md-Hakim/image-colorization-mlops main

lightning_logs/version_42/hparams.yaml

@@ -0,0 +1,7 @@
+display_step: 11
+in_channels: 1
+lambda_gp: 10
+lambda_r1: 10
+lambda_recon: 100
+learning_rate: 2.0e-05
+out_channels: 2

main.py

@@ -53,14 +53,14 @@ except Exception as e:
 
 
 
-STAGE_NAME = 'Model Evaluation Stage'
+# STAGE_NAME = 'Model Evaluation Stage'
 
-try:
-   logger.info(f">>>>>> stage {STAGE_NAME} started <<<<<<") 
-   model_trianer = ModelEvaluationPipeLine()
-   model_trianer.main()
-   logger.info(f">>>>>> stage {STAGE_NAME} completed <<<<<<\n\nx==========x")
-except Exception as e:
-        logger.exception(e)
-        raise e
+# try:
+#    logger.info(f">>>>>> stage {STAGE_NAME} started <<<<<<") 
+#    model_trianer = ModelEvaluationPipeLine()
+#    model_trianer.main()
+#    logger.info(f">>>>>> stage {STAGE_NAME} completed <<<<<<\n\nx==========x")
+# except Exception as e:
+#         logger.exception(e)
+#         raise e
 

requirements.txt

@@ -20,6 +20,10 @@ opencv-python==4.8.0.74
 Pillow==10.0.0
 scikit-image
 tqdm
-
+torchvision
+skimage
+pytorch_lightning
+gc
+scipy