unzipping folder not exist bug fix

README.md

@@ -74,6 +74,8 @@ An deep learning based Image Colorization project.
 <p><small>Project based on the <a target="_blank" href="https://drivendata.github.io/cookiecutter-data-science/">cookiecutter data science project template</a>. #cookiecutterdatascience</small></p>
 
 
+Kaggle API docs:- https://github.com/Kaggle/kaggle-api/blob/main/docs/README.md
+
 ### Version 1:
 
 - im gonna skip logging for now and rather use print statements

kaggle/kernel-metadata.json

@@ -0,0 +1,16 @@
+{
+  "id": "anujpanthri/train-image-colorization",
+  "id_no":345,
+  "title": "training-image-colorization-model",
+  "code_file": "train-image-colorization.ipynb",
+  "language": "python",
+  "kernel_type": "notebook",
+  "is_private": "true",
+  "enable_gpu": "false",
+  "enable_tpu": "false",
+  "enable_internet": "true",
+  "dataset_sources": [],
+  "competition_sources": [],
+  "kernel_sources": [],
+  "model_sources": []
+}

kaggle/train-image-colorization.ipynb

@@ -0,0 +1,55 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!git clone https://github.com/AnujPanthri/Image-Colorization.git\n",
+    "%cd Image-Colorization"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!pip install -r requirements.txt --quiet\n",
+    "!pip install -e ."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!python3 src/scripts/train.py configs/experiment1.yaml"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!python3 src/scripts/visualize_results.py configs/experiment1.yaml"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "name": "python",
+   "version": "3.12.1"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}

src/utils/data_utils.py

@@ -2,7 +2,7 @@ from src.utils.config_loader import constants
 from huggingface_hub import snapshot_download
 from zipfile import ZipFile
 import numpy as np
-import shutil
+import os,shutil
 import matplotlib.pyplot as plt
 import cv2
 import math
@@ -25,6 +25,7 @@ def download_personal_hf_dataset(name):
 def unzip_file(file_path,destination_dir):
     """unzips file to destination_dir"""
     shutil.rmtree(destination_dir)
+    os.makedirs(destination_dir)
     with ZipFile(file_path,"r") as zip:
         zip.extractall(destination_dir)