added assests and sample outputs for README

- cleanup remove `train.py` file - scripts were added - Makefile:: added `deployment cmd`
Muthukamalan · Nov 25, 2024 · fece3f1 · fece3f1
1 parent 8c7ebe2
commit fece3f1
Show file tree

Hide file tree

Showing 8 changed files with 10 additions and 5 deletions.
diff --git a/Makefile b/Makefile
@@ -9,7 +9,7 @@ pull: ## pulling data from s3
 train: ## train mode to check
 	HYDRA_FULL_ERROR=1 python src/train.py 
 	echo "pushing scripted model to s3"
-	aws s3 mv samples/checkpoints/mambaout.pt s3://mhema-dog-breeds-bucket/models/
+	# aws s3 mv samples/checkpoints/mambaout.pt s3://mhema-dog-breeds-bucket/models/
 
 
 test:  ## pytest and code-cov
@@ -81,3 +81,8 @@ format: ## Run pre-commit hooks
 sync: ## Merge changes from main branch to your current branch
 	git pull
 	git pull origin main
+
+
+
+deploy: ## deploy app to huggingface
+	cd gradio/ && gradio deploy
diff --git a/assets/test_confusion_matrix.png b/assets/test_confusion_matrix.png
diff --git a/assets/train_confusion_matrix.png b/assets/train_confusion_matrix.png
diff --git a/assets/val_confusion_matrix.png b/assets/val_confusion_matrix.png
diff --git a/samples/outputs/guess1_prediction.png b/samples/outputs/guess1_prediction.png
diff --git a/samples/outputs/guess2_prediction.png b/samples/outputs/guess2_prediction.png
diff --git a/scripts/schedule.sh b/scripts/schedule.sh
@@ -1,7 +1,7 @@
 #!/bin/bash
 # Schedule execution of many runs
 # Run from root folder with: bash scripts/schedule.sh
-
-python src/train.py trainer.max_epochs=5 logger=csv
-
-python src/train.py trainer.max_epochs=10 logger=csv
+dvc pull
+python src/train.py experiment=hparams logger=csv
+python src/eval.py
+python src/inference.py --input_folder samples/inputs/ --output_folder samples/outputs/ --ckpt_path samples/checkpoints/epoch_019.ckpt 
diff --git a/train.py b/train.py