updated README and added a requirements.txt file

README.md

@@ -5,5 +5,13 @@
 Quick and dirty *Bayesian Network* that classifies pictures of cats and dogs. Works okay-ish..
 
 
-## Installations
+## Running the project (Using VENV)
 
+```sh
+python -m venv .venv
+source .venv/bin/activate
+pip install requirements.txt
+
+python train.py
+python main.py
+```

requirements.txt

@@ -0,0 +1,36 @@
+contourpy==1.3.1
+cycler==0.12.1
+filelock==3.17.0
+fonttools==4.55.6
+fsspec==2024.12.0
+Jinja2==3.1.5
+kiwisolver==1.4.8
+MarkupSafe==3.0.2
+matplotlib==3.10.0
+mpmath==1.3.0
+networkx==3.4.2
+numpy==2.2.2
+nvidia-cublas-cu12==12.4.5.8
+nvidia-cuda-cupti-cu12==12.4.127
+nvidia-cuda-nvrtc-cu12==12.4.127
+nvidia-cuda-runtime-cu12==12.4.127
+nvidia-cudnn-cu12==9.1.0.70
+nvidia-cufft-cu12==11.2.1.3
+nvidia-curand-cu12==10.3.5.147
+nvidia-cusolver-cu12==11.6.1.9
+nvidia-cusparse-cu12==12.3.1.170
+nvidia-nccl-cu12==2.21.5
+nvidia-nvjitlink-cu12==12.4.127
+nvidia-nvtx-cu12==12.4.127
+opencv-python==4.11.0.86
+packaging==24.2
+pillow==11.1.0
+pyparsing==3.2.1
+python-dateutil==2.9.0.post0
+setuptools==75.8.0
+six==1.17.0
+sympy==1.13.1
+torch==2.5.1
+torchvision==0.20.1
+triton==3.1.0
+typing_extensions==4.12.2